Pytorch就地連接並從numpy轉換為張量

Question

假設我有一個張量列表，該張量列表可以沿一個維度連接，例如0。是否有任何命令torch.cat或torch.stack或任何numpy命令在原位進行連接？ 另外，如果我想將numpy ndarray轉換為張量。 如果執行以下操作，在任何給定時間內存中是否有兩個副本？ 我正在處理海量數據集，因此在任何給定時間內存中只能存在一個副本。

# initially data is a huge ndarray
data = torch.Tensor(data)

Answer 1

根據您的評論，假設：

1. 您想要本着以下的精神做一些事情： B = A + a + b + ... + z其中+表示沿着兼容軸的串聯， B和A很大，而a ， b等相對較小，並且
2. 您可以預測B大小的合理上限

我將使用np.empty預先為B分配一個巨大的數組，並根據需要直接用您的數據填充該數組。