tf.cast

tf.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:

x = tf.constant([1.8, 2.2], dtype=tf.float32)
tf.cast(x, 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.