上QQ阅读APP看本书,新人免费读10天
设备和账号都新为新人
3.3 Tensor的形状调整
下面通过两个实例学习Tensor的形状调整。
【例3.5】创建一个二阶张量,长度为6,元素为[0,1,2,3,4,5],使用torch.view()函数将其调整成形状为2×3的Tensor。
输入:
输出:
【例3.6】torch.resize_( )函数是另一种用来调整Tensor形状的方法,但与torch.view()不同,它可以修改Tensor的尺寸。如果新尺寸超过原尺寸,则它会为Tensor自动分配新的内存空间,而如果新尺寸小于原尺寸,则原有的数据依旧会被保存。
输入:
输出: