I would like to index a pytorch tensor with both a boolean mask and normal indices. Something like this: i = 2 j = 0 mask = torch.randn(480, ...
確定! 回上一頁