Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

tf.compat.v1.assert_type

View source on GitHub

Statically asserts that the given Tensor is of the specified type.

tf.compat.v1.assert_type(
    tensor, tf_type, message=None, name=None
)

Args:

  • tensor: A Tensor.
  • tf_type: A tensorflow type (dtypes.float32, tf.int64, dtypes.bool, etc).
  • message: A string to prefix to the default message.
  • name: A name to give this Op. Defaults to "assert_type"

Raises:

  • TypeError: If the tensors data type doesn't match tf_type.

Returns:

A no_op that does nothing. Type can be determined statically.