Boolean indexing of an ndarray with a torch.tensor mask breaks for size ...

Boolean indexing of an ndarray with a torch.tensor mask breaks for size ...

More to explore