pytorh的tensor与numpy的转换,是通过Python buffer protocol完成的。所以这个转换时 zero-copy的。
tensor----》numpy
转换之后的numpy数组,与tensor共用一个基础buffer,所以如果tensor是存储在CPU上的,他们对数据的修改会相互影响。如果tensor被分配在GPU上,
则PyTorch会复制一份数据内容到CPU上。
numpy----》tensor
pytorh的tensor与numpy的转换,是通过Python buffer protocol完成的。所以这个转换时 zero-copy的。
tensor----》numpy
转换之后的numpy数组,与tensor共用一个基础buffer,所以如果tensor是存储在CPU上的,他们对数据的修改会相互影响。如果tensor被分配在GPU上,
则PyTorch会复制一份数据内容到CPU上。
numpy----》tensor