TensorflowでOperatorNotAllowedInGraphError: using a `tf.Tensor` as a Python `bool` is not allowed in Graph execution. Use Eager execution or decorate this function with @tf.function.

tf.data.TFRecordDatasetでmapを使った時に表題のエラー。

mapしたい処理内容を単純化すると以下のような内容。

import tensorflow.experimental.numpy as tnp

2 ** tnp.arange(0,10)

tnp.arange(0,10)をあらかじめインスタンス化することでエラーは出なくなり、map出来るようになった。

temp = tnp.arange(0,10)
2 ** temp

コメント

タイトルとURLをコピーしました