gpt4 book ai didi

pytorch - PyTorch中的唯一张量值

转载 作者:行者123 更新时间:2023-12-04 03:10:43 28 4
gpt4 key购买 nike

我想在PyTorch张量中找到不同的值。
有一些有效的方法来复制Tensorflow的unique op吗?

最佳答案

0.4.0中有一个torch.unique()方法

torch <= 0.3.1中,您可以尝试:

import torch
import numpy as np

x = torch.rand((3,3)) * 10
np.unique(x.round().numpy())

关于pytorch - PyTorch中的唯一张量值,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44993324/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com