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

白小白求教

于 2022-06-20 16:35:34 发布

阅读量671

点赞数 2

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

本文链接：https://blog.youkuaiyun.com/qq_44955721/article/details/125375485

版权

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

2 篇文章

订阅专栏

本文详细介绍了PyTorch中的torch.expand()函数，通过实例展示了如何扩展张量的尺寸。首先创建了一个形状为(1,3)的张量A，接着通过expand()函数将其分别扩展到(3,3)，(-1,3)和(3,-1)。这些操作用于在不同维度上复制张量，以适应不同的计算需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#------------------------------------#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]])