cat

cat((a, b, c), dim)

pytorch中数据的维度依次为(N, C, H, W)

只有被拼接的维度可以不同,其他的必须相同

当在N和C的维度拼接时,直接加在后面即可

当在H维度拼接时,则按行拼接

当在w维度拼接时,则按列拼接

原文地址:https://www.cnblogs.com/liujianing/p/13027174.html