zoukankan      html  css  js  c++  java
  • Pytorch中的torch.cat()函数

    cat是concatnate的意思:拼接,联系在一起。


    先说cat( )的普通用法

    如果我们有两个tensor是A和B,想把他们拼接在一起,需要如下操作:

    C = torch.cat( (A,B),0 )  #按维数0拼接(竖着拼)
    
    C = torch.cat( (A,B),1 )  #按维数1拼接(横着拼)
    >>> import torch
    >>> A=torch.ones(2,3)    #2x3的张量(矩阵)                                     
    >>> A
    tensor([[ 1.,  1.,  1.],
            [ 1.,  1.,  1.]])
    >>> B=2*torch.ones(4,3)  #4x3的张量(矩阵)                                    
    >>> B
    tensor([[ 2.,  2.,  2.],
            [ 2.,  2.,  2.],
            [ 2.,  2.,  2.],
            [ 2.,  2.,  2.]])
    >>> C=torch.cat((A,B),0)  #按维数0(行)拼接
    >>> C
    tensor([[ 1.,  1.,  1.],
             [ 1.,  1.,  1.],
             [ 2.,  2.,  2.],
             [ 2.,  2.,  2.],
             [ 2.,  2.,  2.],
             [ 2.,  2.,  2.]])
    >>> C.size()
    torch.Size([6, 3])
    >>> D=2*torch.ones(2,4) #2x4的张量(矩阵)
    >>> C=torch.cat((A,D),1)#按维数1(列)拼接
    >>> C
    tensor([[ 1.,  1.,  1.,  2.,  2.,  2.,  2.],
            [ 1.,  1.,  1.,  2.,  2.,  2.,  2.]])
    >>> C.size()
    torch.Size([2, 7])

    其次,cat还可以把list中的tensor拼接起来。

    比如:

    上面的代码可以合成一行来写:

  • 相关阅读:
    JS内容左右滑动
    JS返回上一页
    两栏 三栏的css
    舅舅去世
    .net学习开始
    以论坛管理的方式来处理公司资讯
    《尽管去做》摘
    网页视频播放器代码集
    火影忍者和海贼王
    古代风水文献
  • 原文地址:https://www.cnblogs.com/JeasonIsCoding/p/10162356.html
Copyright © 2011-2022 走看看