Мне нужно вернуть, какие элементы целочисленного тензора являются NaN. К сожалению, tf.math.is_nan работает только со следующими типами: bfloat16, half, float32, float64.
Есть ли другой вариант, который не включает преобразование тензора в float и обратно в целое число?
Большое спасибо!