Pytorch 初探

• Pytorch 初探
  • Pytorch简介
  • Pytorch特点
  • 安装Pytorch
  • 基本示例
    • 展示矩阵大小
  • 基本计算方法
  • 索引
  • view操作可以改变矩阵维度
  • 与Numpy的协同操作

Pytorch简介

PyTorch由 Facebook 的 AI 研究团队开发的一个开源的机器学习库，它提供了张量（tensor）计算和动态计算图的功能。

Pytorch特点

安装Pytorch

执行如下命令即可安装Pytorch

# mac or linux
pip install torch torchvision torchaudio
# win
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

安装完成后执行如下命令查看Pytorch版本。

import torch
print(torch.__version__)
print(torch.cuda.is_available())

可以看到我这里没有的gpu只有cpu。

基本示例

创建一个5x3的矩阵。

x = torch.empty(5, 3)
print(x)

创建随机5x3的矩阵。

x = torch.rand(5, 3)
print(x)

初始化一个全零矩阵。

x = torch.zeros(5, 3, dtype=torch.long)
print(x)

直接传入数据。

x = torch.tensor([5.5, 3])
print(x)

创建一个形状为 (5, 3) 的张量，并初始化为全为1

x = x.new_ones(5,3,dtype=torch.double)
# 创建了一个与前面张量相同形状的新张量，但是这次是用从标准正态分布中抽样得到的随机数填充，同时将数据类型设置为 torch.float
x = torch.randn_like(x, dtype=torch.float)
print(x)

展示矩阵大小

x.size()

基本计算方法

做加法。

y = torch.rand(5, 3)
print(x + y)

print(torch.add(x, y))# 一样的也是加法

索引

x

x[:,1]# 取出所有第二个的

x[:,2]# 取出所有第三个的

view操作可以改变矩阵维度

x = torch.randn(4, 4)
y = x.view(16)
z = x.view(-1, 8) # 这里-1表示自动计算
print(x.size(), y.size(), z.size())

x,y,z

m = x.view(-1, 2)
m.size(),m

与-1是相同的。

z = x.view(2, 8)
z.size()

与Numpy的协同操作

numpy与pytorch之间的数据互转。

a = torch.ones(5)
b = a.numpy()
print(a, b)

import numpy as np
a = np.ones(5)
b = torch.from_numpy(a)
b