cat、stack的理解

cat、stack的理解_第1张图片

···for example,在两个tensor中,除了要cat的dim位置的数字可以不一致外,其他位置的数字必须一样。

例如一个老师记录了 4个班级、32个学生、8门课程 【4,32,8】,记a1

另一个老师记录了5个班级、32个学生、8门课程 【5,32,8】,记a2

用torch.cat([a1,a2],dim=0) ,则为【9,32,8】

cat、stack的理解_第2张图片

stack和cat不同的是,会创建一个新的维度,而且要各tensor的shape和size要一样

例如一个老师记录了 一个班级、32个学生、8门课程 【32,8】,记a1

另一个老师记录了另一个班级、32个学生、8门课程 【32,8】,记a2

用torch.stack([a1,a2],dim=0) ,则为【2,32,8】,会在原dim=0 前面 添加一个维度

你可能感兴趣的:(pytorch)