tensorflow和pytorch数据的储存形式

1.tensorflow的数据形式:[B, H, W, C]
2.pytorch的数据形式:[B, C, H, W]

猜你喜欢

转载自blog.csdn.net/xuan_zizizi/article/details/79540106