我有 3 个元素的张量,我想将它们相乘。
我的代码目前看起来像这样:
m1 = tf.multiply(y[0],y[1])
m2 = tf.multiply(m1,y[2])
哪个恕我直言非常不灵活,当然我可以放一个循环并迭代元素,但我想知道 tf 中是否已经提供了这样的功能?我在文档中找不到任何内容
请您参考如下方法:
所需的函数称为 tf.reduce_prod,可在此处找到:https://www.tensorflow.org/api_docs/python/tf/reduce_prod
