您可以调用
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)
您可以调用
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)
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
