首页 文章

了解PyTorch Tensor Shape

提问于
浏览
0

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

input = torch.randn(32, 35)

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

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

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

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

1 回答

  • 0

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

    • 第一个维度可以包含一个元素 .

    • 第二个可以容纳两个 .

    • 第三个可容纳32个元素 .

    • 第四维可以容纳35个元素 .

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

相关问题