1torch.nonzero()找出tensor中非零的元素的索引。返回一个包含输入input中非零元素索引的张量.输出张量中的每行包含input中非零元素的索引。torch.nonzero(input, ...
確定! 回上一頁