tf.cast

tf.cast

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.