torch.expend(-1,3),-1的含义。

最新推荐文章于 2024-06-10 22:18:41 发布

白小白求教

最新推荐文章于 2024-06-10 22:18:41 发布

阅读量554

点赞数 2

分类专栏： pytorch学习记录文章标签： python pytorch

本文链接：https://blog.csdn.net/qq_44955721/article/details/125375485

版权

pytorch学习记录专栏收录该内容

2 篇文章 0 订阅

订阅专栏

#------------------------------------#torch.expend()
A = torch.arange(3).reshape(1,3)
print(A)
A0 = A.size(0) #第0维的尺寸为1
A1 = A.size(1)#第1维的尺寸维3
print(A0)
print(A1)
B = A.expand(3,3)
print(B)
C = A.expand(-1,3)#取当前维度所在的尺寸，现在-1,在0维，尺寸为1
print(C)
D = A.expand(3,-1)#-1在1维，尺寸为3.
print(D)
#------------------------------------#

tensor([[0, 1, 2]])
1
3
tensor([[0, 1, 2], 
        [0, 1, 2], 
        [0, 1, 2]])
tensor([[0, 1, 2]])
tensor([[0, 1, 2], 
        [0, 1, 2], 
        [0, 1, 2]])