numpy 数组到火炬张量转换的工作

Question

我想知道从 NumPy 数组到张量的转换在 PyTorch 和 Tensorflow 中是如何发生的。它会创建整个数组的副本还是有一些就地机制？

其次，如何清除之前被转换为张量的NumPy数组占用的内存。

Answer 1

1. 当device在PyTorch中为CPU时，PyTorch和Numpy在内存中使用相同的n维数组内部表示，所以从Numpy数组转换为PyTorch张量时没有拷贝执行操作时，仅更改它们在内部表示的方式。 Refer here.

2. Python 垃圾收集器使用引用计数来清除未使用的内存，类似于 C++ 中的共享指针。