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]]