PyTorch——(3) tensor基本运算

矩阵乘法

只对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个小的值

在这里插入图片描述

比较运算

在这里插入图片描述在这里插入图片描述

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值