Hi,
Thanks for the answer. Also, how can we get the list of indices corresponding to the “zero rows” ?
Like for above mentioned question it would be like: [0,3]. Thanks!!
edit: I actually want to remove the corresponding columns as well for which rows were zero and removed. So if 1st and 3rd row in a 2D tensor is zero, I want to remove the 1st and 3rd column as well.