Hi guys. Are there any equivalent function for numpy.ma.compress_rowcols in PyTorch where it can suppress the rows and/or columns of a 2-D array that contain NAN value
https://numpy.org/doc/stable/reference/generated/numpy.ma.compress_rowcols.html#numpy.ma.compress_rowcols 2
Of course I can loop through each dims and check for NAN but that will be very inefficient. Thanks for checking by.