标签： python tensorflow machine-learning
我有一个张量A为[m，n，k]，另一个张量B为[m，n]。我想在A的k维上以元素明智的方式添加B。在TensorFlow中有没有一种有效的方法？我一直在寻找broadcast_to，但我认为这不是我想要的。
添加操作是否自动广播？像这样，如果我确实输出= A + B？
谢谢您的帮助！
答案 0 :(得分：1)
tf.add似乎支持广播。需要更仔细地阅读文档！
https://www.tensorflow.org/api_docs/python/tf/math/add