default_collate备忘

# ab_tensor :

( tensor ([ 101, 8024, 3315, ... ]),   

  tensor ([ 1,1,1,... ]), 

  tensor ([ 0,0,0,... ]) )



# ac_tensor :

( tensor ([ 101, 8024, 3315, ... ]),   

  tensor ([ 1,1,1,... ]), 

  tensor ([ 0,0,0,... ]) )



# label_tensor : 

   tensor([1])



# example_tensors :

[ (

     ( tensor ([ 101, 8024, 3315, ... ]), 

       tensor ([ 1,1,1,... ]),   

       tensor ([ 0,0,0,... ]) ),

     ( tensor ([ 101, 8024, 3315, ... ]),   

       tensor ([ 1,1,1,... ]), 

       tensor ([ 0,0,0,... ]) ),

     tensor([1])

   ) ]



res = default_collate( example_tensors )



# res :

[

  [ tensor ([ [ 101, 8024, 3315, ... ] ]),

    tensor ([ [ 1,1,1,... ] ]),   

    tensor ([ [ 0,0,0,... ] ]) ],

  [ tensor ([ [ 101, 8024, 3315, ... ] ]),

    tensor ([ [ 1,1,1,... ] ]),   

    tensor ([ [ 0,0,0,... ] ]) ],

  tensor([[1]])
]

 

对于 tensor ([ 101, 8024, 3315, ... ])，shape: (512)

变为 tensor ([ [ 101, 8024, 3315, ... ] ]), shape: (1, 512)