def predicted_image(self):
# The transpose is because OpenAI gym and chainer have
# different depth conventions
return F.transpose(self._predicted_image[0])
微信
公众号
扫码关注公众号