torch.unbind

Wanderer001

已于 2023-11-02 20:28:41 修改

阅读量4.6k

点赞数 4

分类专栏： Pytorch 文章标签：计算机视觉深度学习机器学习

于 2022-05-14 11:40:06 首次发布

本文链接：https://blog.csdn.net/weixin_36670529/article/details/105255645

版权

Pytorch 专栏收录该内容

193 篇文章 41 订阅

订阅专栏

参考 torch.unbind - 云+社区 - 腾讯云

说明：移除指定维后，返回一个元组，包含了沿着指定维切片后的各个切片。

参数：

tensor(Tensor) -- 输入张量
dim(int) -- 删除的维度

>>> x = torch.randn(3, 3)
>>> x
tensor([[ 0.4775,  0.0161, -0.9403],        
        [ 1.6109,  2.1144,  1.1833],        
        [-0.2656,  0.7772,  0.5989]])
>>> torch.unbind(x, dim=1)(tensor([ 0.4775,  1.6109, -0.2656]), tensor([0.0161, 2.1144, 0.7772]), tensor([-0.9403,  1.1833,  0.5989]))