torch.cat()和torch.stack()的区别,例子解释
torch.stacktorch.stack(inputs, dim=0)inputs:同样也是张量序列,沿着一个新维度对输入张量序列进行连接。 序列中所有的张量都应该为相同形状。torch.cattorch.cat(inputs, dim=0)inputs:必须是张量序列, 在给定维度上对输入的张量序列进行连接操作,序列中所有的张量都应该为相同形状。#注:由字面意思可以看出, c...
原创
2020-04-29 12:20:23 ·
1835 阅读 ·
0 评论