从Tensorflow中的张量中删除一组张量

从Tensorflow中的张量中删除一组张量

您可以调用

tf.unstack

以获取子张量列表。然后，您可以修改列表并调用

tf.stack

以从列表构造张量。例如，以下代码从以下位置删除[2.0，5.0]列：

a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')a_vecs = tf.unstack(a, axis=1)del a_vecs[1]a_new = tf.stack(a_vecs, 1)