|
|
|
@ -19,38 +19,17 @@ namespace paddle {
|
|
|
|
|
class PrintLayer : public Layer {
|
|
|
|
|
public:
|
|
|
|
|
explicit PrintLayer(const LayerConfig& config) : Layer(config) {}
|
|
|
|
|
void forward(PassType passType) override;
|
|
|
|
|
void backward(const UpdateCallback& callback) override {}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
void PrintLayer::forward(PassType passType) {
|
|
|
|
|
Layer::forward(passType);
|
|
|
|
|
for (size_t i = 0; i != inputLayers_.size(); ++i) {
|
|
|
|
|
const auto& argu = getInput(i);
|
|
|
|
|
const std::string& name = inputLayers_[i]->getName();
|
|
|
|
|
if (argu.value) {
|
|
|
|
|
std::ostringstream os;
|
|
|
|
|
argu.value->print(os);
|
|
|
|
|
LOG(INFO) << "layer=" << name << " value matrix:\n" << os.str();
|
|
|
|
|
}
|
|
|
|
|
if (argu.ids) {
|
|
|
|
|
std::ostringstream os;
|
|
|
|
|
argu.ids->print(os, argu.ids->getSize());
|
|
|
|
|
LOG(INFO) << "layer=" << name << " ids vector:\n" << os.str();
|
|
|
|
|
}
|
|
|
|
|
if (auto startPos = argu.sequenceStartPositions) {
|
|
|
|
|
std::ostringstream os;
|
|
|
|
|
startPos->getVector(false)->print(os, startPos->getSize());
|
|
|
|
|
LOG(INFO) << "layer=" << name << " sequence pos vector:\n" << os.str();
|
|
|
|
|
}
|
|
|
|
|
if (auto subStartPos = argu.subSequenceStartPositions) {
|
|
|
|
|
std::ostringstream os;
|
|
|
|
|
subStartPos->getVector(false)->print(os, subStartPos->getSize());
|
|
|
|
|
LOG(INFO) << "layer=" << name << " sub-sequence pos vector:\n"
|
|
|
|
|
<< os.str();
|
|
|
|
|
void forward(PassType passType) override {
|
|
|
|
|
Layer::forward(passType);
|
|
|
|
|
for (size_t i = 0; i != inputLayers_.size(); ++i) {
|
|
|
|
|
getInput(i).printValueString(LOG(INFO),
|
|
|
|
|
"layer=" + inputLayers_[i]->getName() + " ");
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void backward(const UpdateCallback& callback) override {}
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
REGISTER_LAYER(print, PrintLayer);
|
|
|
|
|
|
|
|
|
|