W3cubDocs

/TensorFlow 2.3

tf.sets.size

View source on GitHub

Compute number of unique elements along last dimension of a.

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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.3/api_docs/python/tf/sets/size