如何深入探讨TensorFlow中张量的提取与赋值技巧？

本文共计1323个文字，预计阅读时间需要6分钟。

使用`tf.gather_nd`从`params`中提取特定位置的值，而`tf.scatter_nd`和`tf.scatter_nd_update`则用于更新特定位置的值。`tf.scatter_nd`和`tf.scatter_nd_update`互为逆操作。

tf.gather和gather_nd从params中收集数值，tf.scatter_nd 和 tf.scatter_nd_update用updates更新某一张量。严格上说，tf.gather_nd和tf.scatter_nd_update互为逆操作。

已知数值的位置，从张量中提取数值：tf.gather, tf.gather_nd

tf.gather indices每个元素（标量）是params某个axis的索引，tf.gather_nd 中indices最后一个阶对应于索引值。

tf.gather函数

函数原型

gather( params, indices, validate_indices=None, name=None, axis=0 )

params是要查找的张量，indices是要查找值的索引（int32或int64），axis是查找轴，name是操作名。