觉得有帮助请点赞关注收藏 有问题可评论区留言~~~
Tensor对象是一个维度任意的矩阵，但是一个Tensor中所有元素的数据类型必须一致。torch包含的数据类型和普遍编程语言的数据类型类似，包含浮点型，有符号整型和无符号整形，这些类型既可以定义在CPU上，也可以定义在GPU上。在使用Tensor数据类型时，可以通过dtype属性指定它的数据类型，device指定它的设备（CPU或者GPU）
1：基本定义
代码如下 可查看tensor的基本类型以及运行设备
import torch import numpy as np print('torch.tensor 默认为：{}'.format(torch.Tensor(1).dtype)) a=torch.tensor([[1,2],[3,4]],dtype=torch.float64) print(a) #cpu -gpu之间转换 c=torch.ones((2,2)) c=c.to('cpu',torch.double) print(c.device)
输出如下
2：矩阵乘法
对Tensor执行算术符的运算时 是两个矩阵对应元素的运算，torch.mm执行矩阵乘法的计算
代码如下
import torch import numpy as np #矩阵对应元素运算 a=torch.tensor([[1,2],[3,4]]) b=torch.tensor([[1,2],[4,6]]) c=a*b print("每个元素相乘",c) c=torch.mm(a,b) print("矩阵乘法",c)
3：分段、化整、映射函数
clamp起的是分段函数的作用，可以用于去除矩阵中过小或者过大的元素，round函数将小数部分化整，tanh计算双曲正切函数，该函数将值映射到0,1
代码如下
import torch import numpy as np #特定功能函数 #clamp分段 a=torch.tensor([[1,2],[3,4]]) print(torch.clamp(a,min=2,max=3)) #round化整 a=torch.tensor([[1.1,0.5],[0.6,-1.2]]) print(torch.round(a)) #tanh双曲正切 映射到0 1 print(torch.tanh(a))
四、arange等方法创建数据
除了直接从ndarray或list类型的数据中创建Tensor外，PyTorch还提供了一些函数可以直接创建数据，这类函数往往需要提供矩阵的维度，它和python内置的range的使用方法基本想吐，其第三个参数是步长，linsapce函数第三个参数指定返回的个数。ones返回全1，zeors返回全0
import torch import numpy as np #直接 创建数据 print(torch.arange(5)) print(torch.arange(1,5,2)) print(torch.linspace(0,5,10)) #全0 全1 print(torch.ones(3,3)) print(torch.zeros(3,3))
五、rand返回随机数
rand返回从0 1均匀分布采样的元素所组成的矩阵，randn返回从正态分布采样的元素所组成的矩阵，randint返回指定区间的均匀分布采样的随机整数所组成的矩阵
代码如下
import torch import numpy as np print('torch.tensor 默认为：{}'.format(torch.Tensor(1).dtype)) a=torch.tensor([[1,2],[3,4]],dtype=torch.float64) print(a) #cpu -gpu之间转换 c=torch.ones((2,2)) c=c.to('cpu',torch.double) print(c.device) #矩阵对应元素运算 a=torch.tensor([[1,2],[3,4]]) b=torch.tensor([[1,2],[4,6]]) c=a*b print("每个元素相乘",c) c=torch.mm(a,b) print("矩阵乘法",c) #特定功能函数 #clamp分段 a=torch.tensor([[1,2],[3,4]]) print(torch.clamp(a,min=2,max=3)) #round化整 a=torch.tensor([[1.1,0.5],[0.6,-1.2]]) print(torch.round(a)) #tanh双曲正切 映射到0 1 print(torch.tanh(a)) #直接 创建数据 print(torch.arange(5)) print(torch.arange(1,5,2)) print(torch.linspace(0,5,10)) #全0 全1 print(torch.ones(3,3)) print(torch.zeros(3,3)) #随机 print(torch.rand(3,3)) print(torch.randn(3,3)) print(torch.randint(0,9,(3,3)))
创作不易 觉得有帮助请点赞关注收藏~~~
