TensorFlow 2.0 Beta is available Learn more

tf.sets.size

TensorFlow 2.0 version View source on GitHub

Compute number of unique elements along last dimension of a.

Aliases:

  • tf.compat.v1.sets.set_size
  • tf.compat.v1.sets.size
  • tf.compat.v2.sets.size
  • tf.contrib.metrics.set_size
  • tf.sets.set_size
tf.sets.size(
    a,
    validate_indices=True
)

Args:

  • a: SparseTensor, with indices sorted in row-major order.
  • validate_indices: Whether to validate the order and range of sparse indices in a.

Returns:

int32 Tensor of set sizes. For a ranked n, this is a Tensor with rank n-1, and the same 1st n-1 dimensions as a. Each value is the number of unique elements in the corresponding [0...n-1] dimension of a.

Raises:

  • TypeError: If a is an invalid types.