numpy中有一個函式可以用來匯出兩個張量的并集，如下所示：
import torch
import numpy as np
a = torch.tensor([0, 1, 2])
b = torch.tensor([2, 3, 4])
c = np.union1d(a, b) # c = array([0, 1, 2, 3, 4])
c = torch.from_numpy(c) # c = torch.tensor([0, 1, 2, 3, 4])
但是，我正在尋找可以直接在兩個張量上使用的火炬功能。如果我使用上面的 numpy 函式，我必須將結果從 numpy 轉換為 torch，并且盡管該函式應用于張量，但也必須使用 cpu。
火炬中有沒有可以直接在兩個張量上使用的聯合函式？或者，至少，它可以使用其他 Torch 函式簡單地實作嗎？
uj5u.com熱心網友回復：
您可以使用：
torch.cat((a, b)).unique()
轉載請註明出處，本文鏈接：https://www.uj5u.com/qianduan/511872.html
標籤：麻木的火炬联盟火炬
上一篇：如果N在預定范圍內，則執行操作