torch.stack解析以及与torch.cat的关系

torch.stack这个函数看上去输出的结果很奇怪,但是其实解析之后很简单。 举个例子: 然后使用torch.stack函数: 一眼看上去,好像有点懵逼,不急,我们慢慢解析。 这个函数是这样的,他的dim参数表明他会在dim新建一个维度,然后在这个维度上进行concat。 因此,我们可以这么理解,假如取dim=1,那么我们可以将a,b同时在dim=1新增一个维度,即: 然后再在dim=1的维度上
相关文章
相关标签/搜索