Return the indices of the non-zero elements of the 1-D tensor a as a 1-D tensor.
a
Input
tensor([1, 2, 0, 0, 4, 0])
Output
tensor([0, 1, 4])
Sign in to write and run your own code.
solve(...)