Python深度学习之Pytorch初步使用
作者:Jxiepc 时间:2023-08-12 08:16:41
一、Tensor
Tensor(张量是一个统称,其中包括很多类型):
0阶张量:标量、常数、0-D Tensor;1阶张量:向量、1-D Tensor;2阶张量:矩阵、2-D Tensor;……
二、Pytorch如何创建张量
2.1 创建张量
import torch
t = torch.Tensor([1, 2, 3])
print(t)
2.2 tensor与ndarray的关系
两者之间可以相互转化
import torch
import numpy as np
t1 = np.array(torch.Tensor([[1, 2, 3],
[1, 2, 3]]))
t2 = torch.Tensor(np.array([[1, 2, 3],
[1, 2, 3]]))
运行结果:
2.3 常用api
torch.empty(x,y)
创建x行y列为空的tensor。
torch.ones([x, y])
创建x行y列全为1的tensor。
torch.zeros([x,y])
创建x行y列全为0的temsor。
zeros与empty的区别
后者的数据类型是不固定的。
torch.rand(x, y)
创建3行4列的随机数,随机数是0-1。
torch.randint(low, high, size)
创建一个size的tensor,随机数为low到high之间。
torch.randn([x, y])
创建一个x行y列的tensor,随机数的分布式均值为0,方差1。
2.4 常用方法
item():
获取tensor中的元素,注意只有
一个元素
的时候才可以用。
numpy():
转化成
ndarray
类型。
size()
获取tensor的
形状
。
view()
可以传参,表示获取第几个。若参数为-1,表示不确定,与另一个参数的乘积等于原始形状的乘积。 例如:原始形状为8,则
浅拷贝
,tensor的形状改变
。view(-1,2)
⇒view(4, 2)
; 参数只有-1,表示一维。
dim()
获取维度。
max()
获取最大值。
t()
转置。
transpose(x,y)
x,y是size里面返回的形状相换。
permute()
传入size()返回的形状的顺序。
transpose与permute的区别
前者传入列即可相互交换;后者传入列会根据传入的顺序来进行转化,且需要传入所有列数的索引。
取值[第一阶, 第二阶,……]
一个逗号隔开代表一个阶乘冒号代表全取
赋值[第一阶, 第二阶,……]
直接赋值即可
三、数据类型
3.1 获取数据类型
tensor.dtype
获取数据类型
设置数据类型
注意使用
Tensor()
不能指定数据类型。
type()
修改
数据类型。
四、tensor的其他操作
4.1 相加
torch.add(x, y)
将x和y
相加
。
直接相加
tensor.add()
使用add_()
可相加后直接保存在tensor中
4.2 tensor与数字的操作
tensor + 数值
五、CUDA中的tensor
CUDA
(Compute Unified Device Architecture),是NVIDIA推出的运算平台。CUDATM是一种由NVIDIA推出的通用并行计算架构,该架构使GPU能够解决复杂的计算问题。
torch.cuda这个模块增加了对CUDA tensor的支持,能够在cpu和gpu上使用相同的方法操作tensor通过.to
方法能够把一个tensor转移到另外一个设备(比如从CPU转到GPU)
可以使用torch.cuda.is_available()
判断电脑是否支持GPU
来源:https://blog.csdn.net/weixin_45926547/article/details/116781943
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
基于PyQT5制作一个课堂点名系统
![](https://img.aspxhome.com/file/2023/0/76790_0s.gif)
anaconda3安装及jupyter环境配置全教程
![](https://img.aspxhome.com/file/2023/3/82923_0s.png)
python类的继承实例详解
mysql 查询数据库中的存储过程与函数的语句
设计师的职业规划
Python Flask RESTful使用demo演示
![](https://img.aspxhome.com/file/2023/9/124989_0s.png)
简单了解python单例模式的几种写法
Python制作exe文件简单流程
![](https://img.aspxhome.com/file/2023/2/91852_0s.png)
Python中断多重循环的几种方式详解
python paramiko实现ssh远程访问的方法
python识别验证码的思路及解决方案
![](https://img.aspxhome.com/file/2023/7/64677_0s.png)
Python的GUI框架PySide的安装配置教程
![](https://img.aspxhome.com/file/2023/9/105439_0s.png)
六种酷炫Python运行进度条效果的实现代码
![](https://img.aspxhome.com/file/2023/3/68493_0s.gif)
详解在SpringBoot如何优雅的使用多线程
![](https://img.aspxhome.com/file/2023/9/92069_0s.jpg)
Python中的map()函数和reduce()函数的用法
![](https://img.aspxhome.com/file/2023/0/110360_0s.png)
asp使用 sql_dmo 添加新数据库代码
Python数据相关系数矩阵和热力图轻松实现教程
![](https://img.aspxhome.com/file/2023/4/89434_0s.jpg)
python目录操作之python遍历文件夹后将结果存储为xml
详解Golang中select的使用与源码分析
![](https://img.aspxhome.com/file/2023/8/127388_0s.png)
Django利用AJAX技术实现博文实时搜索
![](https://img.aspxhome.com/file/2023/3/80583_0s.jpg)