Pytorch常用函数用法归纳:创建tensor张量

1.torch.arange()

(1)函数原型

python 复制代码
torch.arange(start,end,step,*,out,dtype,layout=,device,requires_grad)

(2)参数说明:

|---------------|--------------|--------------------------------------------------------------------------------------------------------------------|
| 参数名称 | 参数类型 | 参数说明 |
| start | Number | 起始值,默认值为0 |
| end | Number | 结束值,取不到,为开区间 |
| step | Number | 步长值,默认为1 |
| out | Tensor | 输出的张量,,表明创建后tensor赋予哪个变量,通常情况下不会设置该参数 |
| dtype | torch.dtype | 期望返回的张量的数据类型,如果是None,则使用全局默认值,如果未给出dtype,则从其他输入参数推断数据类型,如果start、end或stop中的任何一个是浮点数,则dtype被推断为默认值,否则被推断为torch.int64 |
| layout | torch.layout | 返回张量的期望 layout,默认值为torch.strided |
| device | torch.device | 返回张量的期望设备。如果是默认值None,则使用当前设备作为默认张量类型,对于CPU类型的张量,则device是CPU;若是CUDA 类型的张量,则device是当前的CUDA 设备 |
| requires_grad | bool | autograd是否记录返回张量上的梯度。默认值为False,表示不记录 |

(3)函数功能:

返回大小为[(end-start)/step]的一维张量,其值为区间**[start,end)**中给定步长为step的等间隔取值

2.torch.range()

(1)函数原型:

java 复制代码
torch.range(start,end,step,*,out,dtype,layout,device=None,requires_grad)

(2)参数说明:

|---------------|--------------|--------------------------------------------------------------------------------------------------------------------|
| 参数名称 | 参数类型 | 参数说明 |
| start | Number | 起始值,默认值为0 |
| end | Number | 结束值,可以取到,为闭区间 |
| step | Number | 步长值,默认为1 |
| out | Tensor | 输出的张量,表明创建后tensor赋予哪个变量,通常情况下不会设置该参数 |
| dtype | torch.dtype | 期望返回的张量的数据类型,如果是None,则使用全局默认值;如果未给出dtype,则从其他输入参数推断数据类型;如果start、end或stop中的任何一个是浮点数,则dtype被推断为默认值;否则被推断为torch.int64 |
| layout | torch.layout | 返回张量的期望layout,默认值为torch.strided,对性能影响不大 |
| device | torch.device | 返回张量的设备。默认值None表示使用当前设备作为默认张量类型;对于CPU类型的张量则device是CPU;若是CUDA 类型的张量则device是当前的CUDA 设备 |
| requires_grad | bool | autograd是否记录返回张量上的梯度。默认值为False,表示不记录 |

(3)函数功能:

返回大小为[(end-start)/step]的一维张量,其值为区间**[start,end]**中给定步长为step的等间隔取值

相关推荐
小宁爱Python7 小时前
基于 Django+Vue3 的 AI 海报生成平台开发(海报模块专项)
人工智能·python·django
红豆怪怪7 小时前
[LeetCode 热题 100] 32. 最长有效括号
数据结构·python·算法·leetcode·动态规划·代理模式
大嘴带你水论文7 小时前
震惊!仅用10张照片就能随意编辑3D人脸?韩国KAIST最新黑科技FFaceNeRF解析!
论文阅读·人工智能·python·科技·计算机视觉·3d·transformer
CodeCraft Studio8 小时前
国产化PDF处理控件Spire.PDF教程:如何在 Java 中通过模板生成 PDF
java·python·pdf·spire.pdf·java创建pdf·从html创建pdf
摆烂z8 小时前
Jupyter Notebook的交互式开发环境方便py开发
ide·python·jupyter
一乐小哥10 小时前
一口气同步10年豆瓣记录———豆瓣书影音同步 Notion分享 🚀
后端·python
华研前沿标杆游学10 小时前
华为在国内搞的研发基地有多野?标杆游学带你解锁“研发界顶流”
python
小胖墩有点瘦10 小时前
【基于深度学习的中草药识别系统】
人工智能·python·深度学习·课程设计·计算机毕业设计·中草药识别
正在走向自律10 小时前
Ubuntu系统下Python连接国产KingbaseES数据库实现增删改查
开发语言·数据库·python·ubuntu·kingbasees·ksycopg2
Calihen的学习日志11 小时前
【Pandas】3.1-数据预处理:列的基本操作
python·pandas