torch.cat

torch.cat

 用于连接多个张量

当dim=0时张量按竖直方向连接，dim=1时按水平方向连接

标签：dim,torch,张量,cat,竖直,连接
来源： https://www.cnblogs.com/hahaah/p/15069822.html