PyTorch——(2) tensor基本操作

最新推荐文章于 2023-12-27 18:45:00 发布

一抹烟霞

最新推荐文章于 2023-12-27 18:45:00 发布

阅读量866

点赞数 1

本文链接：https://blog.csdn.net/qq_34213260/article/details/116136246

版权

维度变换

view()/reshape()等价，参数为需要目标tensor的形状
在这里插入图片描述

在指定位置增加一个维度，参数为增加后，被增加维度所在的位置索引
在这里插入图片描述

只能压缩长度为1的维度，参数为压缩前，被压缩维度的索引
在这里插入图片描述

相当于广播机制，将tensor在“形式上”变成指定的维度。实际数据并没有改变。
在这里插入图片描述 -1表示该维度不扩展

在这里插入图片描述

参数为改变后的顺序
在这里插入图片描述

在这里插入图片描述

在这里插入图片描述

c = torch.rand(4,3,28,32)
# 参数为标量，表示均匀拆分每个成员都分到长度为2
a, b = c.split(2,dim=0)
print(a.shape)
print(b.shape)

在这里插入图片描述

c = torch.rand(4,3,28,32)
# 参数为列表,表示每个成员指定长度
a, b = c.split([1,3],dim=0)
print(a.shape)
print(b.shape)

在这里插入图片描述

参数为需要拆分成的数量

c = torch.rand(4,3,28,32)
a = c.chunk(4,dim=0)
print(a[0].shape)
print(a[1].shape)
print(a[2].shape)
print(a[3].shape)

在这里插入图片描述

关注