【发布时间】:2018-09-02 16:56:06
【问题描述】:
我正在使用动态批量大小进行 DQN 训练。我有两个通行证:
第一关: 具有不同样本的 DNN 的若干后续前向传播,直到发生某些特定事件。在此步骤中,我保存所有输入(它们是一维张量)以在第二遍中重播。因为使用 C++ 我无法创建具有动态形状的张量,所以我将它们存储为一维张量的 std::vector。
第二遍: 计算预期输出并使用相同的 DNN 执行前向传播和反向传播。这次我使用二维张量作为输入,一次计算所有样本。
问题是我如何最有效地从一维张量的向量创建这个二维输入张量?还是我做错了?
【问题讨论】:
标签: c++ tensorflow