numpy:array放在CPU中加速。
torch:Tensor放在GPU中加速。
一.用numpy还是torch
注意numpy、torch中数据转换
二.torch中的数学运算
Torch或numpy
最新推荐文章于 2024-08-10 22:52:59 发布
numpy:array放在CPU中加速。
torch:Tensor放在GPU中加速。
一.用numpy还是torch
注意numpy、torch中数据转换
二.torch中的数学运算