pytorch维度的理解

在学习pytorch实现合并和求和运算时,经常会用到维度的概念,我在第一次接触时还是有点混乱的,在实践操作过程中,总结出一套自己的理解思路,故记录总结到博客上。

此处dim代表维度, 和numpy模块中的axis一样。
对于二维数组而言, dim=0代表行, dim=1代表列。

  • 数组合并操作
    在这里插入图片描述
  • 数组求和操作
    • dim=0代表行相加, 可以理解为所有行相加之和
    • dim=1代表列相加,可以理解为所有列相加之和
      在这里插入图片描述
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值