像tf.reduce_mean和tf.reduce_prod这样的函数执行元素操作以减少沿轴的张量 . 我有一个形状为 (1000, 3, 3)
的张量 R
,一个3x3矩阵的列表 . 我想做的是 matrix 将它们相乘以便我保持一个3x3矩阵 . 如果这是我可以使用的numpy
np.linalg.multi_dot(R)
我怎么能在张量流中做到这一点?
像tf.reduce_mean和tf.reduce_prod这样的函数执行元素操作以减少沿轴的张量 . 我有一个形状为 (1000, 3, 3)
的张量 R
,一个3x3矩阵的列表 . 我想做的是 matrix 将它们相乘以便我保持一个3x3矩阵 . 如果这是我可以使用的numpy
np.linalg.multi_dot(R)
我怎么能在张量流中做到这一点?
1 回答
你可以使用tf.scan:
tf.scan(lambda a, b: tf.matmul(a, b), R)[-1]
这打印: