您的位置:首页 > 其它

tf.cast (API r1.3)

2017-11-18 20:35 411 查看

tf.cast (API r1.3)

1. tf.cast

cast(
x,
dtype,
name=None
)

Defined in tensorflow/python/ops/math_ops.py.

See the guide: Tensor Transformations > Casting

Casts a tensor to a new type.

The operation casts x (in case of Tensor) or x.values (in case of SparseTensor) to dtype.

For example:

# tensor `a` is [1.8, 2.2], dtype=tf.float
tf.cast(a, tf.int32) ==> [1, 2]  # dtype=tf.int32

Args:
x: A Tensor or SparseTensor.
dtype: The destination type.
name: A name for the operation (optional).

Returns:

A Tensor or SparseTensor with same shape as x.

Raises:

TypeError: If x cannot be cast to the dtype.

2. example 1

output:
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: