了解PyTorch张量形状

Question

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

input = torch.randn(32, 35)

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

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

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

我想更好地了解其背后的几何形状。谢谢。

Answer 1

是的，这是正确的。您的input2张量的等级为4。（等级为维度），每个维度的边界为（1,2,32,35）

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

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

Answer 2

将张量形状视为维保存的列表数。例如，张量形状为（1、4、4、2）的张量将包含2个元素中的4个元素中的4个元素。

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