我正在寻找一种简单的方法来从Tensorflow中的当前张量中删除一组张量,并且我有一个难以合理的解决方案.
例如,假设我有以下当前张量:
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
我想删除这个张量中的两个项目(2.0和5.0).
在创建之后将这个张量转换为[1.0,3.0,4.0,6.0]的最佳方法是什么?
提前谢谢了.
解决方法:
您可以调用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)