Is there an efficient way to create a random bit mask in Pytorch? ... mask = torch.randperm(reduce(operator.mul, shape, 1)).float().cuda() ...
確定! 回上一頁