tensorflow - 如何将 tf.int64 转换为 tf.float32？

Question

我尝试过:

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)'

Answer 1

您通常可以使用以下方式进行转换:

tf.cast(my_tensor, tf.float32)

将 tf.float32 替换为您想要的类型。

<小时/>

编辑:至少目前看来，tf.cast不会转换为无符号数据类型(例如tf.uint8) >)。要解决此问题，您可以强制转换为带符号的等效项，并使用 tf.bitcast 来获取所有内容。例如

tf.bitcast(tf.cast(my_tensor, tf.int8), tf.uint8)