UniqueWithCounts

Recherche des éléments uniques le long d'un axe d'un tenseur.

Cette opération renvoie soit un tenseur « y » contenant des éléments uniques le long de « l'axe » d'un tenseur. Les éléments uniques renvoyés sont triés dans le même ordre dans lequel ils apparaissent le long de « l'axe » dans « x ». Cette opération renvoie également un tenseur « idx » et un tenseur « count » qui ont la même taille que le nombre d'éléments dans « x » le long de la dimension « axis ». Le `idx` contient l'index dans la sortie unique `y` et le `count` contient le nombre dans la sortie unique `y`. En d'autres termes, pour un tenseur `1-D` `x` avec `axis = None :

`y[idx[i]] = x[i] pour i dans [0, 1,...,rank(x) - 1]`

Par exemple :

# tensor 'x' is [1, 1, 2, 4, 4, 4, 7, 8, 8]
 y, idx, count = unique_with_counts(x)
 y ==> [1, 2, 4, 7, 8]
 idx ==> [0, 0, 1, 2, 2, 2, 3, 4, 4]
 count ==> [2, 1, 3, 1, 2]
 

# tensor 'x' is [[1, 0, 0],
 #                [1, 0, 0],
 #                [2, 0, 0]]
 y, idx, count = unique_with_counts(x, axis=0)
 y ==> [[1, 0, 0],
        [2, 0, 0]]
 idx ==> [0, 0, 1]
 count ==> [2, 1]
 

# tensor 'x' is [[1, 0, 0],
 #                [1, 0, 0],
 #                [2, 0, 0]]
 y, idx, count = unique_with_counts(x, axis=1)
 y ==> [[1, 0],
        [1, 0],
        [2, 0]]
 idx ==> [0, 1, 1]
 count ==> [1, 2]