pytorch之Torch.split参数的说明

help（torch.split）得到如下形式：简单说就是，split_size_or_sections分为两种形式：int和list

split(tensor, split_size_or_sections, dim=0)解释如下：

   tensor就是设定的输入值，数据类型结构是tensor形式，第二个数字代表，第一个块可以取到原输入第几索引，dim是从第几维度开始取。

split(tensor, split_size_or_sections, dim=0)
    Splits the tensor into chunks.
    
    If :attr:`split_size_or_sections` is an integer type, then :attr:`tensor` will
    be split into equally sized chunks (if possible). Last chunk will be smaller if
    the tensor size along the given dimension :attr:`dim` is not divisible by
    :attr:`split_size`.
    
    If :attr:`split_size_or_sections` is a list, then :attr:`tensor` will be split
    into ``len(split_size_or_sections)`` chunks with sizes in :attr:`dim` according
    to :attr:`split_size_or_sections`.
    
    Arguments:
        tensor (Tensor): tensor to split.
        split_size_or_sections (int) or (list(int)): size of a single chunk or
            list of sizes for each chunk
        dim (int): dimension along which to split the tensor

1、整数型的分块方法
  . 

import torch
x1 = torch.arange(0,192)
x=x1.reshape(4,8,6)
# x =torch.randn(4,8,6)
print('x=\n',x)
y = torch.split(x,3,dim=0)#x:输入；3：在原输入中从零开始取够三个终止；dim=0 是从第0维度切开，也就是4.
y1 = torch.split(x,2,dim=0)
for m in y :
    print('m=\n',m)
    print(m.shape)
for i in y1 :
    print('i=\n',i)
    print(i.size)

x=
 tensor([[[  0,   1,   2,   3,   4,   5],
         [  6,   7,   8,   9,  10,  11],
         [ 12,  13,  14,  15,  16,  17],
         [ 18,  19,  20,  21,  22,  23],
         [ 24,  25,  26,  27,  28,  29],
         [ 30,  31,  32,  33,  34,  35],
         [ 36,  37,  38,  39,  40,  41],
         [ 42,  43,  44,  45,  46,  47]],

        [[ 48,  49,  50,  51,  52,  53],
         [ 54,  55,  56,  57,  58,  59],
         [ 60,  61,  62,  63,  64,  65],
         [ 66,  67,  68,  69,  70,  71],
         [ 72,  73,  74,  75,  76,  77],
         [ 78,  79,  80,  81,  82,  83],
         [ 84,  85,  86,  87,  88,  89],
         [ 90,  91,  92,  93,  94,  95]],

        [[ 96,  97,  98,  99, 100, 101],
         [102, 103, 104, 105, 106, 107],
         [108, 109, 110, 111, 112, 113],
         [114, 115, 116, 117, 118, 119],
         [120, 121, 122, 123, 124, 125],
         [126, 127, 128, 129, 130, 131],
         [132, 133, 134, 135, 136, 137],
         [138, 139, 140, 141, 142, 143]],

        [[144, 145, 146, 147, 148, 149],
         [150, 151, 152, 153, 154, 155],
         [156, 157, 158, 159, 160, 161],
         [162, 163, 164, 165, 166, 167],
         [168, 169, 170, 171, 172, 173],
         [174, 175, 176, 177, 178, 179],
         [180, 181, 182, 183, 184, 185],
         [186, 187, 188, 189, 190, 191]]])
m=
 tensor([[[  0,   1,   2,   3,   4,   5],
         [  6,   7,   8,   9,  10,  11],
         [ 12,  13,  14,  15,  16,  17],
         [ 18,  19,  20,  21,  22,  23],
         [ 24,  25,  26,  27,  28,  29],
         [ 30,  31,  32,  33,  34,  35],
         [ 36,  37,  38,  39,  40,  41],
         [ 42,  43,  44,  45,  46,  47]],

        [[ 48,  49,  50,  51,  52,  53],
         [ 54,  55,  56,  57,  58,  59],
         [ 60,  61,  62,  63,  64,  65],
         [ 66,  67,  68,  69,  70,  71],
         [ 72,  73,  74,  75,  76,  77],
         [ 78,  79,  80,  81,  82,  83],
         [ 84,  85,  86,  87,  88,  89],
         [ 90,  91,  92,  93,  94,  95]],

        [[ 96,  97,  98,  99, 100, 101],
         [102, 103, 104, 105, 106, 107],
         [108, 109, 110, 111, 112, 113],
         [114, 115, 116, 117, 118, 119],
         [120, 121, 122, 123, 124, 125],
         [126, 127, 128, 129, 130, 131],
         [132, 133, 134, 135, 136, 137],
         [138, 139, 140, 141, 142, 143]]])
torch.Size([3, 8, 6])
m=
 tensor([[[144, 145, 146, 147, 148, 149],
         [150, 151, 152, 153, 154, 155],
         [156, 157, 158, 159, 160, 161],
         [162, 163, 164, 165, 166, 167],
         [168, 169, 170, 171, 172, 173],
         [174, 175, 176, 177, 178, 179],
         [180, 181, 182, 183, 184, 185],
         [186, 187, 188, 189, 190, 191]]])
torch.Size([1, 8, 6])
i=
 tensor([[[ 0,  1,  2,  3,  4,  5],
         [ 6,  7,  8,  9, 10, 11],
         [12, 13, 14, 15, 16, 17],
         [18, 19, 20, 21, 22, 23],
         [24, 25, 26, 27, 28, 29],
         [30, 31, 32, 33, 34, 35],
         [36, 37, 38, 39, 40, 41],
         [42, 43, 44, 45, 46, 47]],

        [[48, 49, 50, 51, 52, 53],
         [54, 55, 56, 57, 58, 59],
         [60, 61, 62, 63, 64, 65],
         [66, 67, 68, 69, 70, 71],
         [72, 73, 74, 75, 76, 77],
         [78, 79, 80, 81, 82, 83],
         [84, 85, 86, 87, 88, 89],
         [90, 91, 92, 93, 94, 95]]])
<built-in method size of Tensor object at 0x0000000007895188>
i=
 tensor([[[ 96,  97,  98,  99, 100, 101],
         [102, 103, 104, 105, 106, 107],
         [108, 109, 110, 111, 112, 113],
         [114, 115, 116, 117, 118, 119],
         [120, 121, 122, 123, 124, 125],
         [126, 127, 128, 129, 130, 131],
         [132, 133, 134, 135, 136, 137],
         [138, 139, 140, 141, 142, 143]],

        [[144, 145, 146, 147, 148, 149],
         [150, 151, 152, 153, 154, 155],
         [156, 157, 158, 159, 160, 161],
         [162, 163, 164, 165, 166, 167],
         [168, 169, 170, 171, 172, 173],
         [174, 175, 176, 177, 178, 179],
         [180, 181, 182, 183, 184, 185],
         [186, 187, 188, 189, 190, 191]]])
<built-in method size of Tensor object at 0x00000000078958B8>