PyTorch|一些简单操作

在使用PyTorch构建自己的神经网络时,灵活度非常大,这给了用户更多的发挥空间,这里介绍一些简单的操作。

1,获取所安装PyToch版本

>>> torch.__version__'1.11.0'

2,判断Cuda在我们的系统上是否可用​​​​​​​

>>> torch.cuda.is_available<function is_available >

3,查看张量的形状,shape,size

​​​​​​​

>>> data=torch.tensor([1,2,3,4])>>> data.shapetorch.Size([4])>>> data.size()torch.Size([4])

其中shape是张量的一个属性,而size()是一个方法,所以是对象访问时要带括号。使用dir()函数,可以将一个对象所有方法打印出来,就像这样:

>>> dir(data)

4,dtype,张量的类型

>>> data.dtypetorch.int64

5,device,判断张量在CPU还是GPU​​​​​​​

>>> data.devicedevice(type='cpu')

6,改变维度

改变维度这个操作估计是最令人头痛,比如reshape,flattern,view,squeeze...,对于这些个人感觉最应该的就是进行大量实验直至最后理解不同的参数对应不同的维度。

相关推荐
sp_wxf1 分钟前
Lambda表达式
开发语言·python
DogDaoDao2 分钟前
【预备理论知识——2】深度学习:线性代数概述
人工智能·深度学习·线性代数
牛哥带你学代码3 分钟前
交叠型双重差分法
人工智能·深度学习·机器学习
学步_技术10 分钟前
自动驾驶系列—线控系统:驱动自动驾驶的核心技术解读与应用指南
人工智能·机器学习·自动驾驶·线控系统·转向系统
jmoych19 分钟前
AI正悄然地影响着企业数字化转型
人工智能
蜡笔小新星21 分钟前
Python Kivy库学习路线
开发语言·网络·经验分享·python·学习
说私域21 分钟前
社群团购中的用户黏性价值:以开源小程序多商户AI智能名片商城源码为例
人工智能·小程序
篝火悟者36 分钟前
问题-python-运行报错-SyntaxError: Non-UTF-8 code starting with ‘\xd5‘ in file 汉字编码问题
开发语言·python
hakesashou1 小时前
python如何比较字符串
linux·开发语言·python
深度学习实战训练营1 小时前
基于keras的停车场车位识别
人工智能·深度学习·keras