Tensorflow广播跨越一个维度

Question

我有一个张量a，它是[m，n，k]，还有一个张量B，它是[m，n]。我想在A的k维上以元素的方式添加B。在TensorFlow中有有效的方法吗？我一直在调查broadcast_to，但我不认为这是我想要的。

添加操作是否自动广播？和中一样，如果我明确地输出=A+B？

谢谢你的帮助！

Answer 1

tf.add似乎支持广播。需要更仔细地阅读文档！

https://www.tensorflow.org/api_docs/python/tf/math/add