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

我有一个 [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]] 

谢谢!

请您参考如下方法:

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