aesara.tensor.flatnonzero

aesara.tensor.flatnonzero(a)

Return a vector of indices that are non-zero in the flattened version of a.

Parameters:

a (tensor) – Input tensor

Returns:

Output vector, containing the indices of the elements of a.flatten() that are non-zero.

Return type:

vector

See also

nonzero

Return the indices of the non-zero elements of the input array.

nonzero_values

Return the non-zero elements of the input array