aesara.tensor.nonzero

aesara.tensor.nonzero(a, return_matrix=False)

Returns one of the following:

If return_matrix is False (default, same as NumPy):

A tuple of vector arrays such that the ith element of the jth array is the index of the ith non-zero element of the input array in the jth dimension.

If return_matrix is True (same as Aesara Op):

Returns a matrix of shape (ndim, number of nonzero elements) such that element (i,j) is the index in the ith dimension of the jth non-zero element.

Parameters:

• a (array_like) – Input array.

• return_matrix (bool) – If True, returns a symbolic matrix. If False, returns a tuple of arrays. Defaults to False.

Return type:

tuple of vectors or matrix

See also

nonzero_values

Return the non-zero elements of the input array

flatnonzero

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