而torch.stack拼接的张量维度必须一致为m*n*o...,假设一共有K个张量一起拼接 ... 而np.hstack(),并没有增加维度,一维时,对应于torch.cat(dim = 0){ ...
確定! 回上一頁