如何将矩阵与向量相乘并保留Tensorflow中矩阵的形状？

Question

我知道，在Tensorflow中有一个matmul函数，但是如果我将一个矩阵A与一个向量v相乘，我会得到一个向量w。但是，我需要的是，给定矩阵A：[[a，b]，[c，d]]和向量v：[[e，f]]，我想得到矩阵B：[[ae， bf]，[ce，df]]。矩阵和向量都是张量

Answer 1

您可以使用numpy的multiply功能：

>>> import numpy as np
>>> a = np.array([[1,2],[3,4]])
>>> v = np.array([5,6])
>>> np.multiply(a,v)
array([[ 5, 12],
           [15, 24]])

然后，您可以选择使用.tolist()将numpy数组转换为Python列表：

>>> np.multiply(a,v).tolist()
[[5, 12], [15, 24]]