我有3个元素的张量，我想彼此相乘。
我的代码目前看起来像这样：
m1 = tf.multiply(y[0],y[1])
m2 = tf.multiply(m1,y[2])
哪个imho非常不灵活，当然我可以放一个循环并迭代元素，但我想知道是否已经在tf中提供了这样的功能？我在文档中找不到任何内容
答案 0 :(得分：3)
所需的功能称为tf.reduce_prod，可在此处找到：https://www.tensorflow.org/api_docs/python/tf/reduce_prod
答案 1 :(得分：-1)
使用reduce
import functools
m_final = functools.reduce(lambda x, y: tf.multiply(x, y), your_matrix_list)