@
目录
矩阵乘法
只对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个小的值
比较运算
