最近在学Pytorch,因切片规则和Numpy相同,特在此记录。
多维数组
因为Pytorch中tensor的切片规则和Numpy相同,所以代码中我们用Pytorch来演示
1 | # 随机构建一个三维张量,每个维度有三个元素 |
总结来说,就是张量x
有几个维度,切片操作时x[]
方括号里面就可以有几个下标。:
代表取所有,后面的下标不写就默认是全选,相当于:
积极勤奋,主动勇敢
最近在学Pytorch,因切片规则和Numpy相同,特在此记录。
因为Pytorch中tensor的切片规则和Numpy相同,所以代码中我们用Pytorch来演示
1 | # 随机构建一个三维张量,每个维度有三个元素 |
总结来说,就是张量x
有几个维度,切片操作时x[]
方括号里面就可以有几个下标。:
代表取所有,后面的下标不写就默认是全选,相当于: