常见的张量,无论tf还是torch,都是第一维是行,第二维是列,第三维是通道。 但是在numpy里,第一维是通道,第二维是行,第三维是列。如下图,为numpy的张量: 来源:oschina链接:https://my.oschina.net/u/4135288/blog/4497413 标签 numpy torch