TensorFlow:如何对向量和张量进行点积?

问题描述

我在shape=(n,)的TF中有一个向量(表示一维张量):

my_vector = tf.random.uniform(shape=[n])

我有一个shape=(m,n)的张量:

my_tensor = tf.random.uniform(shape=[m,n])

我想用my_tensormy_vector的每一行进行点积,并从点积中获得shape=(m,)向量。我该怎么办?

我将其元素形象化了。

enter image description here

解决方法

尝试一下

tf.tensordot(my_tensor,my_vector,1)

相关问答

Selenium Web驱动程序和Java。元素在(x,y)点处不可单击。其...
Python-如何使用点“。” 访问字典成员?
Java 字符串是不可变的。到底是什么意思?
Java中的“ final”关键字如何工作?(我仍然可以修改对象。...
“loop:”在Java代码中。这是什么,为什么要编译?
java.lang.ClassNotFoundException:sun.jdbc.odbc.JdbcOdbc...