pytorch乘以4 * 1矩阵和1大小变量发生错误

Question

import torch
from torch.autograd import Variable
import numpy as np

x = np.transpose(np.array([[1, 2, 3, 4]]))
a = Variable(torch.rand(1), requires_grad=True)

print(a * x) # error!

如果a = 2，我想要x = [[2] [4] [6] [8]]这样的结果

有什么解决办法吗？

Answer 1

您正在寻找的是矩阵乘法中的点标量积。

尝试：

x = np.transpose(np.array([[1, 2, 3, 4]]))
a = 2
x.dot(a)

输出矩阵[[2] [4] [6] [8]]