什么是PYTORCH ?
这是一个基于Python的科学计算软件包,针对两组受众:
- NumPy的替代品,可以使用GPU的强大功能
- 深入学习研究平台,提供最大的灵活性和速度
什么是PYTORCH ?
这是一个基于Python的科学计算软件包,针对两组受众:
NumPy的替代品,可以使用GPU的强大功能
深入学习研究平台,提供最大的灵活性和速度
入门
张量
张量与NumPy的ndarray类似,另外还有Tensors也可用于GPU以加速计算。
from __future__ import print_function
import torch
注意
声明了未初始化的矩阵,但在使用之前不包含明确的已知值。创建未初始化的矩阵时,当时分配的内存中的任何值都将显示为初始值。
构造一个未初始化的5x3矩阵:
x = torch.empty(5, 3)
print(x)
tensor([[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.],
[0., 0., 0.]])
构造一个随机初始化的矩阵:
x = torch.rand(5, 3)
print(x)
tensor([[0.6259, 0.0797, 0.8297],
[0.6732, 0.7944, 0.2363]