【Python】reshape里的-1
reshape里的-1
·
一句话:-1在哪个维度,执行reshape的时候自动计算哪个维度。
举例:首先生成一个4*4的Tensor:
import torch
x = torch.arange(16)
x=x.reshape(4,4)
print(x)
tensor([[ 0, 1, 2, 3],
[ 4, 5, 6, 7],
[ 8, 9, 10, 11],
[12, 13, 14, 15]])
这是大家都知道的。现在我们把第一个维度指定为-1,第二个维度指定为2.因为Tensor里面一共16个元素,那么第一个维度自动计算出16/2=8.(如果不是整数则报错)。
因此,这里reshape(-1,2)就等价于reshape(8,2)。
x=x.reshape(-1,2)
print(x)
tensor([[ 0, 1],
[ 2, 3],
[ 4, 5],
[ 6, 7],
[ 8, 9],
[10, 11],
[12, 13],
[14, 15]])
现在我们把两个维度交换一下,那么输出就变成两行。此时reshape(2,-1)等价于reshape(2,8):
x=x.reshape(2,-1)
print(x)
tensor([[ 0, 1, 2, 3, 4, 5, 6, 7],
[ 8, 9, 10, 11, 12, 13, 14, 15]])
特别的,reshape(1,-1)转化成一行,reshape(-1,1)转化成一列。

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。
更多推荐
所有评论(0)