提交 258e5ec5 编写于 作者: L liaogang

update GradientMachine API

上级 7c5fd231
...@@ -146,7 +146,7 @@ void GradientMachine::randParameters() { m->machine->randParameters(); } ...@@ -146,7 +146,7 @@ void GradientMachine::randParameters() { m->machine->randParameters(); }
Matrix* GradientMachine::getLayerOutput(const std::string& layerName) const Matrix* GradientMachine::getLayerOutput(const std::string& layerName) const
throw(UnsupportError) { throw(UnsupportError) {
auto nn = std::dynamic_pointer_cast<paddle::NeuralNetwork>(m->machine); auto nn = m->machine;
if (nn) { if (nn) {
auto mat = nn->getLayerOutput(layerName); auto mat = nn->getLayerOutput(layerName);
return Matrix::createByPaddleMatrixPtr(&mat); return Matrix::createByPaddleMatrixPtr(&mat);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册