pytorch官方网站
PyTorch是一个基于Python的科学计算软件包:numpy的替代品,可使用强大的GPU功能;一个深度学习平台,具有最大的灵活性和最快的速度。
-
张量(Tensors)
与numpy的ndarrays类似,用于GPU加速计算。一个未初始化的5行3列矩阵
a = torch.empty(5,3) print(a)
输出为
一个随机初始化的5行3列矩阵a = torch.rand(5, 3) print(a)
输出为
一个5行3列,值均为0的矩阵,类型为longa = torch.zeros(5, 3, dtype=torch.long) print(a)
输出为
一个直接给定数值的张量a = torch.tensor([5.5, 3]) print(a)
输出为