.transpose函数的作用
images = raw_float.reshape([-1, num_channels, img_size, img_size])
images = images.transpose([0, 2, 3, 1])
这里images.transpose([0, 2, 3, 1])是将images的shape变成[num,img_size, img_size,num_channels ]
欢迎使用Markdown编辑器
你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Mar
原创
2020-05-11 23:53:09 ·
1891 阅读 ·
0 评论