らんだむな記憶

blogというものを体験してみようか!的なー

TF 2.0 tensor.numpy() inside of map() functions

TF 2.0 tensor.numpy() inside of map() functions · Issue #30035 · tensorflow/tensorflow · GitHubに遭遇した。

print(tf.constant(3).numpy())

tf.data.Dataset.map の内外で呼んでみると、内部では exception が飛ぶ。
なんか、tf.constant() の動きもそうだが、map の外では tensorflow.python.framework.ops.EagerTensor としてテンソルが作成されるものの、map の中では tensorflow.python.framework.ops.Tensor になっているように見える。そしてこの状態だと NG という感じで。
同じく map の中ではApplications - Keras Documentationにあるような tf.keras.applications.vgg19.preprocess_input もうまく適用できないように見える。どうもこの APIEagerTensor でないと作用できないようだ。map の中では使える API が限られている?専用のものを使う必要がある?など何かありそうだがよく分からない・・・。