什么是PyTorch?
PyTorch是一个基于Python的科学计算库,它有以下特点:
- 类似于NumPy,但是它可以使用GPU
- 可以用它定义深度学习模型,可以灵活地进行深度学习模型的训练和使用
Tensors
Tensor类似与NumPy的ndarray,唯一的区别是Tensor可以在GPU上加速运算。
In [1]:
from __future__ import print_function import torch
构造一个未初始化的5x3矩阵:
PyTorch是一个基于Python的科学计算库,它有以下特点:
Tensor类似与NumPy的ndarray,唯一的区别是Tensor可以在GPU上加速运算。
In [1]:
from __future__ import print_function import torch
构造一个未初始化的5x3矩阵: