tensorflow有没有类似pytorch的“masked_fill_”的功能

Question

我想通过掩码矩阵将 INF 值设置为矩阵，就像 pytorch 代码一样：

scores.data.masked_fill_(y_mask.data, -float('inf'))

我尝试使用tf.map_fn它来实现，但性能太慢。那么tensorflow有没有高效的函数来实现呢？

Answer 1

我用数学计算方法来代替。它是有效的并且速度更快。

def mask_fill_inf(matrix, mask):
    negmask = 1 - mask
    num = 3.4 * math.pow(10, 38)
    return (matrix * mask) + (-((negmask * num + num) - num))

有人有更好的方法吗？