tf.cast(
x,
dtype,
name=None
)
tf.cast可以改變tensor的數(shù)據(jù)類型躬存。
x是我們要操作的tensor张惹,而dtype是我們要將tensor變成的數(shù)據(jù)類型。
舉例如下:
x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.cast(x, tf.int32) # [1, 2], dtype=tf.int32
完成了將一個數(shù)據(jù)類型為float32得tensor變?yōu)閕nt32
類型的tensor的操作岭洲。