import torch
import numpy as np
from scipy.stats import rankdata
a=torch.tensor([0, 2, 3, 2])
b=np.array([0, 2, 3, 2])
rankdata(a, method='dense')#tensor
rankdata(b, method='dense')#numpy都行
array([1, 2, 3, 2], dtype=int32)
其他用法如下:
参考:https://vimsky.com/zh-tw/examples/usage/python-scipy.stats.rankdata.html。