python - torch.unique 不适用于浮点张量

Question

我正在尝试从浮点张量中提取独特的元素。我试过了 ：

out = torch.unique(my_tensor)

但是，此方法仅适用于 int/long 张量。我的张量是以非均匀方式量化的张量，因此它保证具有一小组浮点值。

标签： pythonpytorch

Answer 1

你可以numpy.unique改用

import torch
import numpy as np

t = torch.tensor([1.05, 1.05, 2.01, 2.01, 3.9, 3.9001])
print(np.unique(t.numpy()))

输出：

[1.05   2.01   3.9    3.9001]