I have a  84x84  pytorch tensor named  target . I need to mask it with an  84x84  boolean numpy array which consists of  True  and  False . This mask array is called mask.
When I do  target = target[mask] , I get the error  TypeError: can't convert np.ndarray of type numpy.bool_. The only supported types are: double, float, float16, int64, int32, and uint8.
Surprisingly, I get this error only when running on a GPU. When running on a CPU, everything works fine. How can I fix this?