5. Indices of Non-Zero Elements

Easy

Return the indices of the non-zero elements of the 1-D tensor a as a 1-D tensor.

Implement solve(...)