维度超出范围（预计在 [-2, 1] 范围内，但得到 2）

Question

为什么会弹出以下错误？什么应该在这个范围内？为什么？-2维是什么意思？

RuntimeError: dimension out of range (expected to be in range of [-2, 1], but got 2)

这段代码会产生错误

import torch 

torch.bmm(torch.randn(1000, 784) , torch.randn(784, 10))

Answer 1

方法torch.bmm实现了批量矩阵-矩阵乘积。对于普通矩阵-矩阵乘积，您需要两个有两个二维矩阵才能创建乘积。

\n\n

和torch.bmm可以批量创建产品，但当然您需要包含批量维度，因此您需要两个输入 3 维矩阵。

\n\n

关于如何使用尺寸torch.bmm：

\n\n

\n
如果batch1是(b\xc3\x97n\xc3\x97m)张量，\nbatch2 是( b\xc3\x97m\xc3\x97p)张量，输出将为(b\xc3\x97n\xc3\x97p)张量。
\n

\n\n

\n