我有一个可能很简单的 Tensorflow 问题。我试图弄清楚当我的掩码是 1D 时如何执行 2D boolean_mask。我有一个具有 N X F 维的 2D 张量 x 和一个具有 N 维的一维张量 mask。 mask 包含 0 和 1，我想将 x 中 mask 值为 1 的样本保留。
threshold = tf.multiply(tf.ones_like(mask), 0.5)
mask = tf.greater(mask,threshold)
x = tf.boolean_mask(x,mask)
由于形状不兼容，我收到值错误。有没有简单的方法可以解决这个问题？
最佳答案
其中 N 是我想要扩展的功能数量。
threshold = tf.multiply(tf.ones_like(mask), 0.5)
mask = tf.greater(mask,threshold)
x = tf.boolean_mask(x,tf.tile(mask,[1,N]))
关于python - Tensorflow 中的 2D bool 掩码，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/44478149/