假设我有一个相同大小的张量列表,它们可以在一个维度上连接,比如0.执行任何命令torch.cat或torch.stack或任何numpy命令就地进行连接?另外,如果我想将numpy ndarray转换为tensor。如果我执行以下操作,在任何给定时间内存储器中是否存在两个副本?我正在处理一个庞大的数据集,所以在任何给定的时间内,只有一个副本可以存在于内存中。
# initially data is a huge ndarray
data = torch.Tensor(data)
答案 0 :(得分:-1)
从您的评论中,假设:
B = A + a + b + ... + z
其中+
表示沿着兼容轴的连接,B
和A
是巨大的a
,b
等,相对较小,B
我会事先使用B
为np.empty
分配一个巨大的数组,我会根据需要直接用数据填充此数组。