Function GpuRNNGetParamCopy

src/model/operation/rnn.cc:507–519

}

Tensor GpuRNNGetParamCopy(int linLayerID, int pseudoLayer, Tensor &weights,
                          bool is_bias, CudnnRNNHandle &h) {
  size_t offset, size;
  std::tie(offset, size) =
      h.weights_mapping[std::make_tuple(linLayerID, pseudoLayer, is_bias)];
  Tensor paramCopy(
      Shape{
          size,
      },
      weights.device());
  CopyDataToFrom(&paramCopy, weights, size, 0, offset);
  return paramCopy;
}

/*
vector<Tensor> GpuRNNForwardTrainingEx();