Hi @ptrblck,
How do I convert a 2d mask to index  while retaining the dimensions.
like if my mask is of shape Nx1000. How do i use this to create a index tensor of size (N,) as each sample in the batch may not have the same number of masked elements.
Thanks