张量索引张量与张量切片

张量索引张量与张量切片

提示:
张量的索引与一般的列表和元组索引类似,在不同维度的使用索引值索引,与一般列表相似的数字操作可以参考 https://blog.csdn.net/Zzz_zhongqing/article/details/107279857
广播机制 https://www.cnblogs.com/jiaxin359/p/9021726.html

这里用一个张量或者列表作为索引参数来索引张量,如 a [ b ] ,这里做为索引的张量 b,它的数据类型必须是long,byte,或者bool型的。
张量索引与列表索引十分类似
以 a = torch.randn(2, 3, 4)为例子,a 的 size 是 2 * 3 * 4 .

a
tensor([[[ 0.9347,  1.6853,  0.1960,  0.6464],
         [-0.5048,  0.8314,  0.8814, -1.3094],
         [-1.2213,  0.3590, -1.0769,  0.8755]],

        [[-0.0164, -0.2710, -1.6122,  0.5343],
         [-1.1255, -0.8615, -0.2898,  0.0288],
         [-0.3857<
  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值