在药典中创建切片蒙版？

Question

有没有办法根据切片操作指定掩码？

例如

A = torch.arange(6).view((2,3))

# A = [[0,1,2], [3,4,5]]

mask_slice = torch.mask_slice(A[:,1:])

# mask_slice = [[0,1,1],[0,1,1]]

Answer 1

你可以做这样的事情（如果我答对了你的问题）：mask_slice = torch.zeros(A.shape, dtype=bool)mask_slice[:, 1:] = 1# tensor([[False,  True,  True],#         [False,  True,  True]])