【基操】torch.stack的使用场景

1. 多个 tensor 合并成一个大tensor

   if isinstance(inputs, list):
   	inputs = torch.stack(inputs)
   

   例如：inputs 是一个list， 其中有几个相同维度的 tensor，那么可以将其使用stack，合并成更高维度的 tensor。