1. ホーム
  2. tensorflow

[解決済み] tf.int64をtf.float32に変換する方法は?

2022-02-08 05:17:26

質問

試してみました。

test_image = tf.convert_to_tensor(img, dtype=tf.float32)

すると、次のようなエラーが表示されます。

ValueError: Tensor conversion requested dtype float32 for Tensor with dtype int64: 'Tensor("test/ArgMax:0", shape=TensorShape([Dimension(None)]), dtype=int64)'

解決方法は?

おっと、APIに関数があった...。

 tf.to_float(x, name='ToFloat')