在Pytorch中实现了Numpy

我们使用x[:, None]用于元分矩阵乘法。
如:
x = np.array([1.8507 ,-2.7324])
y = np.array([0.9722 , 0.4470 , 1.0000 , 0.0000 , 0.0000 , 0.0000])
如果我们这样做,x * y我们会得到错误:
ValueError: operands could not be broadcast together with shapes (2,) (6,)
但是x * y[:,None],会给我们
array([[ 1.79925054, -2.65643928],
[ 0.8272629 , -1.2213828 ],
[ 1.8507 , -2.7324 ],
[ 0. , -0. ],
[ 0. , -0. ],
[ 0. , -0. ]])
是否有pytorch这种功能的实现?
已邀请:
匿名用户

匿名用户

赞同来自:

你可以试试unsqueeze

要回复问题请先登录注册