PyTorch 1.0 中文官方教程:什么是 PyTorch

2022-05-07 13:57:54 浏览数 (1)

译者:bat67 最新版会在译者仓库首先同步。

PyTorch是一个基于python的科学计算包,主要针对两类人群:

  • 作为NumPy的替代品,可以利用GPU的性能进行计算
  • 作为一个高灵活性、速度快的深度学习平台

入门

张量

Tensor(张量)类似于NumPyndarray,但还可以在GPU上使用来加速计算。

代码语言:javascript复制
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]])

阅读全文/改进本文

0 人点赞