tf.cast可以改变tensor的数据类型。
tf.cast(
x,
dtype,
name=None
)
eg:
x = tf.constant([2.1, 2.1], dtype=tf.float32)
y=tf.cast(x, tf.int32) # [2, 2], dtype=tf.int32
with tf.Session() as sess:
print(sess.run(x))
print(sess.run(y))
tf.cast可以改变tensor的数据类型。
tf.cast(
x,
dtype,
name=None
)
eg:
x = tf.constant([2.1, 2.1], dtype=tf.float32)
y=tf.cast(x, tf.int32) # [2, 2], dtype=tf.int32
with tf.Session() as sess:
print(sess.run(x))
print(sess.run(y))