pytorch快速入门，源码免费下载

安装pytorch

pytorch环境的搭建可以很复杂，也可以很简单。因为只是学习阶段，所以这里咱们采用最简单的搭建方式，基于CPU的搭建。

我的电脑的操作系统是Windows11，具体的安装步骤如下。

安装：

pip install ipython

pip3 install numpy --pre torch torchvisiontorchaudio --force-reinstall --index-url https://download.pytorch.org/whl/nightly/cpu

验证安装：

import torch

print(torch.__version__)

创建张量

就像Numpy有Array，pandas有Series和DataFrame一样，PyTorch也有自己非常重要的一种基础数据结构张量。创建张量的方式主要如下图所示。

示例：创建特殊形状的张量

import torch

# 创建张量

a = torch.Tensor([[1, 2], [3, 4]])

print(a)

print(a.type())

# 指定张量的形状

b = torch.Tensor(2, 3)

print(b)

# 全是1的张量

c = torch.ones(2, 3)

print(c)

# 全是0的张量

d = torch.zeros(2, 3)

print(d)

e = torch.zeros_like(a)

print(e)

示例：创建随机数据的张量

import torch

# 随机值

a = torch.rand(2, 3)

print(a)

# 正态分布

b = torch.normal(mean=0, std=torch.rand(5))

print(b)

# 序列

c = torch.arange(0, 10, 2)

print(c)

# 等差数列

d = torch.linspace(2, 22, 8)

print(d)

代码下载

代码下载地址：pytorch_basic_01.zip:https://url22.ctfile.com/f/38921122-907983357-c13790?p=3553 (访问密码: 3553)