有沒有辦法根據(jù)切片操作指定掩碼？例如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]]
1 回答
胡說叔叔
TA貢獻1804條經(jīng)驗 獲得超8個贊
你可以做這樣的事情（如果我答對了你的問題）：
mask_slice = torch.zeros(A.shape, dtype=bool)
mask_slice[:, 1:] = 1
# tensor([[False, True, True],
# [False, True, True]])
添加回答
舉報
0/150
提交
取消