摘要: 写代码的时候想把一个张量\(X\)中的最后一个维度进行类似集合那样的操作,于是网上找到了torch.unique这个方法(官方文档) torch.unique(input, sorted=True, return_inverse=False, return_counts=False, dim=Non 阅读全文
posted @ 2023-09-22 17:35 kksk43 阅读(425) 评论(0) 推荐(0)
特效
黑夜
侧边栏隐藏