2.6 与numpy交互

时间:2020-07-30 13:22:04   收藏:0   阅读:67

pytorh的tensor与numpy的转换,是通过Python buffer protocol完成的。所以这个转换时 zero-copy的。

 

tensor----》numpy

技术分享图片

 

 转换之后的numpy数组,与tensor共用一个基础buffer,所以如果tensor是存储在CPU上的,他们对数据的修改会相互影响。如果tensor被分配在GPU上,

则PyTorch会复制一份数据内容到CPU上。

 

numpy----》tensor

技术分享图片

 

原文:https://www.cnblogs.com/superxuezhazha/p/13402921.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!