Определение того, транслируются ли тензоры tf.cast внутри некоторого кода тензорного потока? ⇐ Python

[Anonymous]

Можно ли программно определить, включает ли выполнение некоторого кода тензорного потока приведение каких-либо тензоров, например с tf.float32 на tf.float64? Недавно я удалил некоторый код, который делал это без необходимости в сложной функции, и хотел бы протестировать его (а также убедиться, что нет других случаев, которые можно было бы исправить).

Например, тест проверит, что вызов bad_double с float32 включает приведение, а вызов good_double — нет:
def bad_double(x: tf.Tensor) -> tf.Tensor: return tf.cast(tf.cast(x, dtype=tf.float64) * 2, x.dtype) def Good_double(x: tf.Tensor) -> tf.Tensor: вернуть х * 2