python - 通过 tensorflow 输出二维 boolean 掩码值

Question

我有一个 [n,m] 输入和 [n,m] boolean 掩码。我怎么能输出一个 [n,x] 过滤矩阵，而不是 [x] 数组？
例如输入

x = [[1,2,3],[4,5,6]]

和一个 boolean 掩码

bm = [[1,0,1],[1,0,0]]

我尝试使用 tf.boolean_mask() 并得到 [1,3,4]。我怎样才能得到像

这样的二维结果

result =  [[1,3],[4]]

谢谢!

Answer 1

TF 中不存在此操作，因为 TF 对张量进行操作并返回张量。您收到的结果包含不同数量的元素，而不是张量。