Torch.stack() 1. 概念 在一个新的维度上连接一个张量序列 2. 参数 tensors (sequence)需要连接的张量序列 dim (int)在第dim个维度上连接 注意输入的张量shape要完全一致,且dim必须小于len(tensors). 3. 举例 3.1 四个shape为[3, 3]的张量 以下面这4个张量,每个张量shape为[3, 3]. 1 a = torch.Tensor([[1,2,3],[4,5,6],[7,8,9]]) 2 b = torch.Tenso…