はtensorflowは、私はちょうどpytorchのコードのように、マスク行列で行列にINF値を設定したいpytorchの「masked_fill_」

Question

と同様の機能を持っています遅すぎる。それで、テンソルフローはそれを実装するための効率的な関数を持っていますか？

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

もっと良い方法がありますか？