1. 程式人生 > >TensorFlow中張量轉置操作tf.cast/tf.dtypes.cast用法詳解

TensorFlow中張量轉置操作tf.cast/tf.dtypes.cast用法詳解

一、環境

TensorFlow API r1.12

CUDA 9.2 V9.2.148

Python 3.6.3

二、官方說明

tf.cast 或 tf.dtypes.cast

將輸入張量轉換資料型別

tf.dtypes.cast(
    x,
    dtype,
    name=None
)

輸入:

(1)x:可以是張量(Tensor)、稀疏張量(SparseTensor)或數值型別的索引切片(IndexedSlices),如:uint8, uint16, uint32, uint64, int8, int16, int32, int64, float16, float32, float64, complex64, complex128, bfloat16.

(2)dtype:要轉換的目標資料型別,支援的資料型別種類同上面的x

(3)name:可選引數,操作的名稱

三、示例

>>> x = tf.constant([1.8,2.2])
>>> x1 = tf.cast(x,tf.int32)
>>> with tf.Session() as sess:
...     print(sess.run(x1))
...
[1 2]