了解PyTorch Tensor Shape

Question

我有一个关于我们在PyTorch中定义的张量形状的简单问题。如果我说：

input = torch.randn(32, 35)

这将创建一个包含32行和35列的矩阵。现在我定义：

input2 = torch.randn(1,2,32, 35)

关于新矩阵输入2的维数，我能说些什么？如何在此处定义行和列？我的意思是我有两个由张量包装的形状为32 * 35的矩阵？

我想更好地理解这背后的几何。谢谢。

Answer 1

对，那是正确的。你的input2张量的等级为4.（Rank是维度），每个维度的界限是（1,2,32,35）

1. 第一个维度可以包含一个元素。
2. 第二个可以容纳两个。
3. 第三个可容纳32个元素。
4. 第四维可以容纳35个元素。

编辑：我发现将高维数组视为一系列列表是有用的。在您的情况下，等级4张量，将是列表列表的列表。