Note. 当前,在CUDA实现和CPU实现中,当指定dim时,无论 sort 参数如何, torch.unique 总是在开始时对张量进行排序。排序可能很慢,因此,如果您的输入张量已被排序, ...
確定! 回上一頁