torch.nonzero(..., as_tuple=True) returns a tuple of 1-D index tensors, ... Returns a tensor containing the indices of all non-zero elements of input .
確定! 回上一頁