pytorch乘以4*1矩阵和1个size变量出现错误

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!

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

有什么解决办法吗？

Answer 1

你要找的是矩阵乘法中的点标量积。

尝试：

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

这输出一个矩阵[[2][4][6][8]]

pytorch乘以4*1矩阵和1个size变量出现错误

pytorch multiply 4*1 matrix and 1 size variable occur error

python

pytorch