第40个方法torch.split(tensor, split_size_or_sections, dim=0)此方法和前面的torch.chunk()一样 ... pytorch每日一学48(torch.vstack())垂直(行方向)按顺序堆叠张量.
確定! 回上一頁