会员
周边
众包
新闻
博问
闪存
所有博客
当前博客
我的博客
我的园子
账号设置
简洁模式
...
退出登录
注册
登录
Picassooo
博客园
首页
新随笔
联系
订阅
管理
2024年7月15日
pytorch的expand函数
摘要: PyTorch 中的 expand 函数用于扩展张量的形状,使其在某些维度上“看起来”像被复制了多次,但实际上它不会复制数据,从而节省内存和计算资源。扩展后的张量共享原始张量的内存空间,因此原始张量和扩展后的张量是同一个数据的视图。 以下是 torch.expand 函数的一些基本用法: 1. 扩展
阅读全文
posted @ 2024-07-15 09:22 Picassooo
阅读(10)
评论(0)
推荐(0)
编辑
公告