newbie__jixiang的博客

原创 【PyTorch】介绍及操作

在我看来，PyTorch就是一个能使用GPU加速矩阵运算的工具库，因为卷积在特征提取中的大量运用使得矩阵运算变得越来越多、也需要运算的速度变得越快越好。我们知道CPU擅长于复杂的运算，但核心数较少更适合于串行计算；而GPU核心数多但对于复杂运算的计算能力较弱，所以更适合于单个任务较简单的并行计算。这些特点十分符合大量卷积运算的需求，所以使用GPU加速能极大地提升我们的运算速度！