如何在Torch中扩展dims（向张量添加维度）
2016-11-29 1:4:20
收藏：0
阅读：137
评论：1
我想向张量添加维度，就像 numpy.expand_dims 那样。我该怎么做？
我可以使用以下代码实现：
a = torch.Tensor({{1,2}, {3,4}, {5,6}}) --shape (3,2)
a_size_table = a:size():totable()
table.insert(a_size_table, 1, 1) -- 在第一维之前添加`1`
a:reshape(torch.LongStorage(a_size_table)) -- 返回形状为 (1,3,2) 的张量
这对我来说似乎太麻烦了（与 np.expand_dims(a,0) 相比）。也许有更好的方法？
点赞
评论区的留言会收到邮件通知哦~
作者:
用户4744283
推荐文章
- Lua 虚拟机加密load(string.dump(function)) 后执行失败问题如何解决
- 我想创建一个 Nginx 规则，禁止访问
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1：1导入真实世界的地形？
- 求解，lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教：如何用lua实现游戏内调用数据库函数实现账号密码注册？
- Lua实现网站后台开发
- LUA错误显式返回，社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接，并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下，有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值（不考虑键的顺序）
- 有个lua简单的项目，外包，有意者加微信 liuheng600456详谈，最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码？
有一个
nn.utils.addSingletonDimension函数，我在这里找到了答案：https://www.bountysource.com/issues/38773074-convenience-function-to-add-a-singleton-dimension-to-a-torch-tensor。