FlatnonzeroΒΆ
Return indices that are non-zero in the flattened version of a.
Abstract Signature:
Flatnonzero(a: Array)
PyTorch
API:
βStrategy: Macro 'torch.nonzero({a}.flatten(), as_tuple=True)[0]'
Return indices that are non-zero in the flattened version of a.
Abstract Signature:
Flatnonzero(a: Array)
β