如何在不添加图的情况下将张量对象转换为数组?

时间:2019-06-20 21:47:44

标签: python tensorflow keras

我基本上有一个for循环,我需要在其中运行预测才能计算出一些东西。

for i in range(0,6,1):
        episodes, n_actions = util.record_episode(env,num = 2)
        data, actions, targets = util.forward_data(episodes)
        new_image, pred_image, action = latent_model.predict([(data - np.mean(data, axis=0)) / 255])
        latent_actions = tf.Variable(models.argmin_mse(new_image,np.moveaxis(np.repeat(np.array([targets]),4,axis = 0),0,1)))

但是,当我播放大量剧集时,我最终会耗尽内存。我试图使计算图静态化,以查看是否发生了有趣的事情。事实证明,“ tf.Variable”正在向图表添加内容吗?

有人可以帮助我将张量转换为numpy数组而不会耗尽内存吗?

0 个答案:

没有答案