如何将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不会转换为无符号的dtype(例如tf.uint8).要解决此问题,您可以转换为已签名的等效项并用于tf.bitcast获取所有权限.例如

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

Answer 2

哎呀,我在API中找到了这个函数......

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

Answer 3

您可以使用其中任何一个tf.cast(x, tf.float32)或tf.to_float(x)两个,转换为float32.

例:

sess = tf.Session()

# Create an integer tensor.
tensor = tf.convert_to_tensor(np.array([0, 1, 2, 3, 4]), dtype=tf.int64)
sess.run(tensor)
# array([0, 1, 2, 3, 4])

# Use tf.cast()
tensor_float = tf.cast(tensor, tf.float32)
sess.run(tensor_float)
# array([ 0.,  1.,  2.,  3.,  4.], dtype=float32)

# Use tf.to_float() to cast to float32
tensor_float = tf.to_float(tensor)
sess.run(tensor_float)
# array([ 0.,  1.,  2.,  3.,  4.], dtype=float32)