tshtsha的博客

原创 Python-torch.cat()拼接操作

A=torch.ones(2,2,3,3) dim负数就是从后往前 -1 -2 -3 -4 从前往后0 1 2 3。A=torch.ones(2,2,3,3) 中的dim取值为[-4,3]今天在阅读源码，发现dim为负数，很难理解。