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)
相关推荐
redreamSo3 分钟前
大模型是不是到顶了?瓶颈到底在哪
人工智能·openai
用户8356290780516 分钟前
使用 Python 管理 Excel 工作表:创建、复制、删除与重命名
后端·python
Oo9207 分钟前
Tool Use 背后的技术逻辑
人工智能
姗姗来迟了9 分钟前
Vue3封装AI流式对话组件踩坑实录
人工智能
码上天下1 小时前
用Pinia管理AI多会话状态
人工智能
用户054324329702 小时前
Next.js接大模型流式SSE实操踩坑
人工智能
Assby2 小时前
从 Function Calling 到 MCP:理解 Agent 工具调用的底层通信机制
人工智能·后端
小星AI2 小时前
Claude Code 从入门到精通,一步到位
人工智能
后端小肥肠2 小时前
Codex + Obsidian 做人生副本视频:输入主题文案,直通剪映草稿
人工智能·aigc·agent
百度Geek说3 小时前
全链路研发智能体 ——从"体感能用"到"实际可用"的工程实践
人工智能