首页 文章

从std :: vector <Tensor>值创建Tensor

提问于
浏览 次
1

我正在进行动态批量大小的DQN培训 . 我有两个通行证：

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

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

问题是如何从一维张量的向量中最有效地创建这个二维输入Tensor？或者我做错了吗？

c++ tensorflow

1 回答

  • 0

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

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

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

    tensor[i][j]

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

    回复于

相关问题