flipped_images = images.flip([3])

在代码实现中发现, images.flip([3]) 是使得图像沿着X轴随机进行翻转。即这里的3代表的是X轴。
因为images.shape=[128,9,84,84]。因此,这里的3表示的 wide 的值。
同理,如果需要沿着2旋转,即沿着Y轴进行旋转,也是可以的。

posted @ 2022-06-23 18:00  呦呦南山  阅读(52)  评论(0编辑  收藏  举报