网站建设平台市场重庆定制网站开发价格
在conda的环境中安装Jupyter及其他软件包
Pytorch 建立在张量（tensor）之上，Pytorch张量是一个 n 维数组，类似于 NumPy 数组。专门针对GPU设计，可以运行在GPU上以加快计算效率。换句话说，Pytorch张量是可以运行在GPU上的多维数据。
目录
基础语法
定义张量数据类型
默认数据类型
指定数据类型
改变张量数据类型
张量与Numpy数组间的转换
将张量转换为 Numpy 数组
将 Numppy 数组转换为张量
在设备间移动张量
默认在CPU上定义张量
定义CUDA设备
将张量移动到CUDA设备
将张量移动到CPU
在任意设备上定义张量
基础语法
| Pytorch张量 | Numpy数组 | |
|---|---|---|
| torch.ones(.) | numpy.ones(.) | 创建一个一数组 |
| torch.zeros(.) | numpy.zeros(.) | 创建一个零数组 |
| torch.rand(.) | numpy.random.rand(.) | 创建随机数组 |
| torch.tensor(.) | numpy.array(.) | 从给定值创建数组 |
| x.shape or x.size() | x.shape | 获取数组形状 |
定义张量数据类型
默认数据类型
默认张量数据类型为 torch.float32
import torch
x = torch.ones(2, 2)
print(x)
print(x.dtype) 
指定数据类型
x = torch.ones(2, 2, dtype=torch.int8)
print(x)
print(x.dtype)
改变张量数据类型
先定义一个张量，打印其数据类型
x=torch.ones(1)
print(x.dtype)
通过".type()"命令改变数据类型
x=x.type(torch.torch.uint8)
print(x.dtype)
可以看到数据类型由torch.float32改变为了torch.uint8
张量与Numpy数组间的转换
将张量转换为 Numpy 数组
定义一个随机张量
x=torch.rand(2,2)
print(x)
print(x.dtype) 
通过".numpy()"命令转换为numpy数组
y=x.numpy()
print(y)
print(y.dtype)
将 Numppy 数组转换为张量
定义一个numpy数组
import numpy as np
x=np.zeros((2,2),dtype=np.float32)
print(x)
print(x.dtype)
用".from_numpy()"命令转换为张量
y=torch.from_numpy(x)
print(y)
print(y.dtype)
在设备间移动张量
一般情况下，Pytorch 张量存储在 CPU 上，张量运行在 GPU上 以加快计算速率，这是张量的主要优势。为了有效利用这个优势，需要将张量移动到 CUDA 设备，可以使用 .to 方法将张量移动到相应设备上。
默认在CPU上定义张量
使用".tensor()"命令，张量一般默认定义在CPU上
x=torch.tensor([1, 1.5])
print(x)
print(x.device)
定义CUDA设备
if torch.cuda.is_available():device=torch.device("cuda:0")
torch.cuda.current_device() 
将张量移动到CUDA设备
x = x.to(device)
print(x)
print(x.device)
将张量移动到CPU
device = torch.device("cpu")
x = x.to(device)
print(x)
print(x.device)
在任意设备上定义张量
device = torch.device("cuda:0")
x = torch.ones(2,2, device=device)
print(x)