译者:bat67 最新版会在译者仓库首先同步。
PyTorch是一个基于python的科学计算包,主要针对两类人群:
- 作为NumPy的替代品,可以利用GPU的性能进行计算
- 作为一个高灵活性、速度快的深度学习平台
入门
张量
Tensor
(张量)类似于NumPy
的ndarray
,但还可以在GPU上使用来加速计算。
from __future__ import print_function
import torch
创建一个没有初始化的5*3矩阵:
代码语言:javascript复制x = torch.empty(5, 3)
print(x)
输出:
代码语言:javascript复制tensor([[2.2391e-19, 4.5869e-41, 1.4191e-17],
[4.5869e-41, 0.0000e 00, 0.0000e 00],
[0.0000e 00, 0.0000e 00, 0.0000e 00],
[0.0000e 00, 0.0000e 00, 0.0000e 00],
[0.0000e 00, 0.0000e 00, 0.0000e 00]])
阅读全文/改进本文