3 #include "tensorflow/core/platform/env.h" 4 #include "tensorflow/core/public/session.h" 10 {
"input_slice",
"vars_slice" }
14 {
"input_png2d",
"vars_png2d" },
15 {
"input_png3d",
"vars_png3d" }
19 "target_primary",
"target_total" 24 : simpleModel(savedir, sliceInputs, prongInputs,
outputs)
31 const float primaryE = outputs[0].tensor<
float,2>()(0, 0);
32 const float totalE = outputs[1].tensor<
float,2>()(0, 0);
static const std::vector< std::string > outputs
static const std::vector< InputConfigKeys > sliceInputs
LSTMEnergy predict(const VarDict &varDict)
Model(const std::string &savedir)
static const std::vector< InputConfigKeys > prongInputs
std::vector< tensorflow::Tensor > predict(const VarDict &varDict) const