如何解决如何从张量流中的张量中删除特定值的向量?
我要执行以下操作。 给定张量,
m = ([[1,1,1],[2,2,2],[3,3,3]])
如何实现从m中删除值[2,2,2]的向量?
解决方法
您可以这样做:
import tensorflow as tf
def remove_row(m,q):
# Assumes m is 2D
mask = tf.math.reduce_any(tf.not_equal(m,q),axis=-1)
return tf.boolean_mask(m,mask)
# Test
m = tf.constant([[1,1,1],[2,2,2],[3,3,3]])
q = tf.constant([2,2])
tf.print(remove_row(m,q))
# [[1 1 1]
# [3 3 3]]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。