PyTorch是一个基于python的科学计算包,主要针对两类人群:
Tensor
(张量)类似于NumPy
的ndarray
,但还可以在GPU上使用来加速计算。
from __future__ import print_function
import torch
创建一个没有初始化的5*3矩阵:
x = torch.empty(5, 3)
print(x)
输出:
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]])