Python 如何从tensorflow中的张量中删除一个特定值的向量？

我想实现以下操作。 给定张量

m = ([[1, 1, 1], [2, 2, 2], [3, 3, 3]])

---

如何实现从m中删除值为[2,2,2]的向量？

您可以这样做：

将tensorflow导入为tf
def移除_行（m，q）：
#假设m是2D
mask=tf.math.reduce_any（tf.not_equal（m，q），axis=-1）
返回tf.boolean_掩码（m，掩码）
#试验
m=tf.常数（[[1,1,1]，[2,2,2]，[3,3,3]]
q=tf.常数（[2,2,2]）
tf.打印（删除第（m，q）行）
# [[1 1 1]
#  [3 3 3]]

我刚刚测试了您的解决方案，太好了。非常感谢。@wangsy很高兴能帮上忙。如果你觉得答案解决了你的问题，请考虑把它标记为被接受。