从std :: vector创建一个张量 <Tensor> 价值观

Question

我正在使用动态批次大小进行DQN培训。 我有两张通行证：

第一遍：DNN的几个后续后续传播带有不同的样本，直到发生特定事件为止。 在此步骤中，我保存所有输入（一维张量）以在第二遍中重播。 因为使用C ++无法创建具有动态形状的Tensor，所以将它们存储为一维张量的std :: vector。

第二遍：计算预期输出，并使用相同的DNN进行正向传播和反向传播。 这次我使用二维张量作为输入，一次计算所有样本。

问题是如何最有效地从一维张量向量创建二维输入张量？ 还是我做错了一切？

Answer 1

您可以从向量的向量定义张量！ 定义一个向量，该向量的每个元素也是一个向量。 例如：

 vector<int> tensor(i,vector<int> (j)); //the size is i*j where i and j are integers

要访问ij的元素，您可以通过以下方式访问它：

tensor[i][j]

由于向量的大小是动态的，因此该张量的大小是动态的