广州专业网站建设性价比高58百度搜索引擎
一、torch概念
1.1简介
pytorch简称torch，意为深度学习框架。它使用张量（tensor）来表示数据，可以轻松地处理大规模数据集，且可以在GPU上加速。
pytorch基本功能：自动微分、自动求导等。
1.2安装
官网获得下载命令。
二、tensor概述
pytorch将数据封装成张量进行计算。
2.1概念
张量：元素类型相同的多维矩阵，可以在GPU上于运行。
张量的维度决定了它的形状：
（1）0维张量：标量；
（2）1维张量：向量、数组；
（3）2维张量：矩阵；
（4）更高维度张量：图像、视频等。
2.2特点
（1）动态计算图；
（2）GPU支持；
（3）自动微分。
2.3数据类型
主要数据类型：浮点数float，整数int，布尔值True、False。
其中，浮点数和整数又分为8位、16位、32位、64位，加起来共9种。
三、tensor创建
3.1基本创建方式
（1）torch.tensor()
注意tensor为小写，根据指定的数据创建张量。
代码示例：
# 指定数据进行创建 torch.tensor
import torch
import numpy as np
# 用标量创建张量
data = torch.tensor(7) #0维张量
print(data)
print(data.shape)结果：
tensor(7)
torch.Size([])
（2）torch.Tensor()
注意tensor为大写，根据形状创建张量，也可用指定的数据创建张量。
代码示例：
# 指定形状进行创建 torch.Tensor
def test01():tensor1 = torch.Tensor(3,4)print(tensor1)tensor2 = torch.Tensor([[1],[2],[3]])print(tensor2,tensor2.dtype)tensor3 = torch.Tensor([0])print(tensor3.shape,tensor3.dtype)tensor1 = torch.Tensor([1,2]).int()print(tensor1.dtype)if __name__ == '__main__':test01()结果：
tensor([[-4.6947e-33, 1.9716e-42, 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]])
tensor([[1.], [2.], [3.]]) torch.float32
torch.Size([1]) torch.float32
torch.int32
（3）torch.IntTensor()
用于创建指定类型的张量。
代码示例：
# 指定类型进行创建 torch.IntTensor
data = torch.IntTensor(3,3)
print(data)
print(data.shape,data.dtype)结果：
tensor([[-1983708176, 1407, 0],
[ 0, 0, 0],
[ 0, 0, 0]], dtype=torch.int32)
torch.Size([3, 3]) torch.int32
3.2创建线性张量
利用API：
torch.arange():PyTorch 中用于生成一个一维张量（向量）的函数，该张量包含从起始点到结束点（不包括结束点）的等间隔数值序列。
torch.linspace((start,end,steps)):返回一个一维张量，包含在区间 `[start, end]` 上均匀间隔的 `steps` 个点。
代码示例：
# 创建线性张量
import torch
def tensor():data = torch.arange(0,9,2)print(data,data.shape,data.dtype)data1 = torch.linspace(1,10,3)print(data1,data1.shape,data1.dtype)tensor()
结果：
tensor([0, 2, 4, 6, 8]) torch.Size([5]) torch.int64
tensor([ 1.0000, 5.5000, 10.0000]) torch.Size([3]) torch.float32
3.3创建随机张量
随机数种子：torch.manual_seed()
获取随机数种子：torch.initial_seed()
创建随机张量：torch.rand--均匀分布 / torch.randn--正态分布
代码示例：
# 创建随机张量
import numpy as np
def random_t():torch.manual_seed(33)print(torch.initial_seed())data = torch.rand(3,4)data2 = torch.randn(2,3).int()print(data,data.shape,data.dtype)print(data2,data2.shape,data2.dtype)return
random_t()结果：
33
tensor([[0.6186, 0.5587, 0.1937, 0.3360],
[0.2008, 0.6970, 0.6367, 0.1452],
[0.8213, 0.2365, 0.3702, 0.9168]]) torch.Size([3, 4]) torch.float32
tensor([[ 0, 0, -1],
[ 0, -1, 1]], dtype=torch.int32) torch.Size([2, 3]) torch.int32
四、tensor常见属性
4.1获取属性
Tensor张量常见属性有三个：dtype（获取类型）,shape（获取形状）,device（获取运行设备）
代码示例：
import torch
import numpy as np
data = torch.tensor(np.random.randint(0,10,(2,2)))
print(data)
print(data.shape)
print(data.dtype)
print(data.device)结果：
tensor([[4, 4], [6, 5]], dtype=torch.int32)
torch.Size([2, 2])
torch.int32
cpu
4.2切换设备
主要是指CPU，GPU之间的相互转换。
注意：这里都要重新赋值
（1）直接data.cuda()
（2）在创建张量时指定device类型
（3）重新赋值data.to(device)
代码示例：
# 切换设备
import torch
import numpy as np
data = torch.tensor(np.random.randint(0,10,(2,2)))
print(data)
print(data.shape)
print(data.dtype)
print(data.device)device = 'cuda' if torch.cuda.is_available() else 'cpu'
data_new = data.to(device=device)
print(data_new.device)data2 = torch.tensor([1,2,3],device='cpu')
print(data2.device)data3 = torch.tensor([1,2,3],dtype=torch.float32)
data3_new = data3.cuda()
print(data3_new.device)结果：
tensor([[4, 6], [5, 8]], dtype=torch.int32)
torch.Size([2, 2])
torch.int32
cpu
cpu
cpu
Torch not compiled with CUDA enabled(这里是因为我没有安装GPU)
4.3类型转换
（1）使用type进行转换，在type（）括号内填上需要转换为的目标类型；
（2）重新赋值 data = data.int()
（3）创建张量时指定类型 dtype= torch.int64
代码示例：
# 类型转换
import torch
import numpy as npdata = torch.tensor(np.random.randn(3,4),dtype=torch.int64)
print(data,data.dtype)data2 = torch.tensor([[1,2],[3,4]])
data2 = data2.float()
print(data2,data2.dtype)data3 = torch.tensor([8,9,0])
data3 = data3.type(torch.bool)
print(data3,data3.dtype)结果：
tensor([[ 0, 0, 0, -2], [ 0, 1, 0, -1], [ 1, 0, 0, 0]]) torch.int64
tensor([[1., 2.], [3., 4.]]) torch.float32
tensor([ True, True, False]) torch.bool
五、小结
综上，简单学习了张量的创建以及张量相关属性的获取，数据类型的修改方式。都是一些简单的基本操作，但是对于不熟的代码还是要勤加练习。