는 pytorch의 "masked_fill_"

Question

난 그냥 pytorch 코드처럼, 마스크 행렬로 행렬하는 INF 값을 설정할 유사한 기능이 tensorflow합니까 너무 느립니다. 그렇다면 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)) 

더 좋은 방법이 있습니까?