Я делю два тензора, которые включают нули, друг с другом.
test = torch.zeros(10000)
div = torch.div(test, test) # 99497.19it/s
Однако результат div включает nan с, и использование этого выходного тензора портит градиент прилично, поэтому я хочу заполнить div запасным значением 0.
div[div != div] = 0 # 28546.27it/s
Эта операция исправления тензора очень затратна и изменяет всю последовательность с запуска 100 000 раз в секунду до 5500.
Есть ли более быстрый способ либо разделить с помощью запасного значения, либо заполнить все nan с помощью 0?