Pytorch就地连接并从numpy转换为张量

时间:2018-06-11 14:26:22

标签: python numpy pytorch

假设我有一个相同大小的张量列表,它们可以在一个维度上连接,比如0.执行任何命令torch.cat或torch.stack或任何numpy命令就地进行连接?另外,如果我想将numpy ndarray转换为tensor。如果我执行以下操作,在任何给定时间内存储器中是否存在两个副本?我正在处理一个庞大的数据集,所以在任何给定的时间内,只有一个副本可以存在于内存中。

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

1 个答案:

答案 0 :(得分:-1)

从您的评论中,假设:

  1. 您希望本着以下精神做一些事情:B = A + a + b + ... + z其中+表示沿着兼容轴的连接,BA是巨大的ab等,相对较小,
  2. 您可以预测B
  3. 大小的合理上限

    我会事先使用Bnp.empty分配一个巨大的数组,我会根据需要直接用数据填充此数组。