Pytorch torch.normal()的用法

该函数原型如下:

复制代码
normal(mean, std, *, generator=None, out=None)

该函数返回从单独的正态分布中提取的随机数的张量,该正态分布的均值是mean,标准差是std。

用法如下:我们从一个标准正态分布N~(0,1),提取一个2x2的矩阵

复制代码
torch.normal(mean=0.,std=1.,size=(2,2))

我们也可以让每一个值服从不同的正态分布,我们还是生成2x2的矩阵:

复制代码
torch.normal(mean=torch.arange(4.),std=torch.arange(1.,0.6,-0.1)).reshape(2,2)
相关推荐
我的username19 分钟前
极致简单的openclaw安装教程
人工智能
小锋java123425 分钟前
【技术专题】嵌入模型与Chroma向量数据库 - Chroma 集合操作
人工智能
七月丶25 分钟前
别再手动凑 PR 了:这个 AI Skill 会按仓库习惯自动建分支、拆提交、提 PR
人工智能·设计模式·程序员
AI攻城狮34 分钟前
用 Playwright 实现博客一键发布到稀土掘金
python·自动化运维
用户5191495848451 小时前
CVE-2024-10793 WordPress插件权限提升漏洞利用演示
人工智能·aigc
曲幽1 小时前
FastAPI分布式系统实战:拆解分布式系统中常见问题及解决方案
redis·python·fastapi·web·httpx·lock·asyncio
chaors1 小时前
从零学RAG0x01之向量化
人工智能·aigc·ai编程
chaors1 小时前
从零学RAG0x02向量数据库
人工智能·aigc·ai编程
陈少波AI应用笔记1 小时前
硅谷龙虾大战技术拆解:当AI长出爪子
人工智能