PyTorch——(3) tensor基本运算

@

目录

• 矩阵乘法
• tensor的幂
• exp()/log()
• 近似运算
• clamp() 截断
• norm() 范数
• max()/min() 最大最小值
• mean() 均值
• sun() 累加
• prod() 累乘
• argmax()/argmin() 最大最小值所在的索引
• topk() 取最大的n个
• kthvalue() 第k个小的值
• 比较运算

矩阵乘法

只对2d矩阵有效，因为矩阵乘法是定义在2d矩阵上的

torch.mm(a,b)
torch.matmul(a,b)
a@b

tensor的幂

对tensor中的每一个元素都进行幂运算

exp()/log()

近似运算

clamp() 截断

将小于10的都等于10

将大于10的都等于10

norm() 范数

参数为几范数

max()/min() 最大最小值

mean() 均值

sun() 累加

prod() 累乘

argmax()/argmin() 最大最小值所在的索引

- 指定维度

将结果的维度和输入的数据维度一致

keepdim = True

topk() 取最大的n个

max()相当于topk(1)

largest=True 表示最大的k个 
largest = False 表示最小的k个

kthvalue() 第k个小的值

比较运算