TensorFlow中如何仅对张量的指定列做加法并保留其余列原数据

直接通过张量拼接操作即可满足需求，完整可运行代码如下：

import tensorflow as tf

test = tf.constant([[[100., 2., -30.],[-4,5,6]], [[4., 5., 6.],[-7,8,9]]])
test1 = tf.constant([[[100.]],[[8.]]])

# 核心逻辑：前两列加test1后，和原始最后一列沿最后一个维度拼接
res = tf.concat([test[..., :2] + test1, test[..., 2:]], axis=-1)
print(res)

运行后输出和你给出的预期结果完全一致。这里用...是TensorFlow的省略索引写法，可以自动匹配前面所有维度，写法更简洁不易出错。

内容的提问来源于stack exchange，提问作者freak11