flipped_images = images.flip([3])

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