... and add a batch dimension (BCHW) return resize(screen).unsqueeze(0) env.reset() plt.figure() plt.imshow(get_screen().cpu().squeeze(0).permute(1, 2, ...
確定! 回上一頁