TensorFlow 2.0 RC is available Learn more

tf.compat.v2.debugging.assert_type

View source on GitHub

Asserts that the given Tensor is of the specified type.

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

This can always be checked statically, so this method returns nothing.

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 for this operation. Defaults to "assert_type"

Raises:

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