tensorflow - 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))

谁有更好的方法？