torch.unique()和np.unique()的转换

筱阳^_^

已于 2024-12-13 22:21:55 修改

阅读量273

点赞数 2

文章标签： python pytorch numpy

于 2024-12-13 16:59:24 首次发布

本文链接：https://blog.csdn.net/weixin_46995912/article/details/144454903

版权

torch.unique()功能说明见链接

np.unique()功能说明见链接

torch.unique(A, return_inverse=True)和np.unique(A, return_inverse=True)是等价的

这里想补充一点差异，

a, indices = np.unique(A, return_index=True)   # 返回新列表元素在旧列表中的位置（下标）

这个功能torch.unique()是没有的，因此考虑另一种实现方式：

def unique(x, dim=0):
    unique, inverse, counts = torch.unique(x, dim=dim, 
        sorted=True, return_inverse=True, return_counts=True)
    decimals = torch.arange(inverse.numel(), device=inverse.device) / inverse.numel()
    inv_sorted = (inverse+decimals).argsort()
    tot_counts = torch.cat((counts.new_zeros(1), counts.cumsum(dim=0)))[:-1]
    index = inv_sorted[tot_counts]
    index = index.sort().values
    return unique, inverse, counts, index