tf.cast()数据类型转换

tf.cast()数据类型转换函数

tf.cast()函数的作用是执行tensorflow中张量数据类型转换，将图片数据类型为int8类型的，转换为float32。

cast定义:
cast(x,dtype,name=None)

x：被转换的数据(张量)
dtype：目标数据类型
name：可选参数，定义操作名称

int32转换为float32:

import tensorflow as tf
import numpy as np

t1=tf.Variable([1,2,3,4,5])
t2=tf.cast(t1,dtype=tf.float32)

print('t1:{}'.format(t1))
print('t2:{}'.format(t2))

输出结果：
t1:<tf.Variable ‘Variable:0’ shape=(5,) dtype=int32, numpy=array([1, 2, 3, 4, 5])>
t2:[1. 2. 3. 4. 5.]