TensorFlow 张量投射
2018-09-08 15:28 更新
tf.cast
cast ( x , dtype , name = None )
定义在:tensorflow/python/ops/math_ops.py
参见指南:张量变换>铸造
将张量投射到新的类型.
该操作将 x(在张量的情况下)或 x.values(在 SparseTensor 情况下)投射到 dtype.
例如:
#tensor`a` is [1.8,2.2],dtype = tf.float tf.cast(a, tf.int32 ) == > [ 1 , 2 ] #dtype = tf.int32
ARGS:
- x:一个张量或 SparseTensor.
- dtype:目标类型.
- name:操作的名称(可选).
返回:
与 x 具有相同形状的张量或 SparseTensor.
注意:
- TypeError:当 x 不能投射到 dtype.