Package ai.djl.paddlepaddle.engine

Class PpSymbolBlock

  • All Implemented Interfaces:
    ai.djl.nn.Block, ai.djl.nn.SymbolBlock
    public class PpSymbolBlock
extends ai.djl.nn.AbstractSymbolBlock
    PpSymbolBlock is the PaddlePaddle implementation of SymbolBlock.

    • Field Summary

      • Fields inherited from class ai.djl.nn.AbstractBlock

        children, inputShapes, parameters, version

    • Method Summary

      All Methods Instance Methods Concrete Methods 
      Modifier and Type Method Description
      protected ai.djl.ndarray.NDList forwardInternal​(ai.djl.training.ParameterStore parameterStore, ai.djl.ndarray.NDList inputs, boolean training, ai.djl.util.PairList<java.lang.String,​java.lang.Object> params)
      ai.djl.ndarray.types.Shape[] getOutputShapes​(ai.djl.ndarray.types.Shape[] inputShapes)

      • Methods inherited from class ai.djl.nn.AbstractBlock

        addChildBlock, addParameter, beforeInitialize, cast, clear, describeInput, forward, forward, forwardInternal, getChildren, getDirectParameters, getParameters, initialize, initializeChildBlocks, isInitialized, loadMetadata, loadParameters, prepare, readInputShapes, saveInputShapes, saveMetadata, saveParameters, setInitializer, setInitializer, setInitializer, toString

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

      • Methods inherited from interface ai.djl.nn.Block

        cast, clear, describeInput, forward, forward, forward, getChildren, getDirectParameters, getParameters, initialize, isInitialized, loadParameters, saveParameters, setInitializer, setInitializer, setInitializer

      • Methods inherited from interface ai.djl.nn.SymbolBlock

        describeOutput, removeLastBlock

    • Constructor Detail

      • PpSymbolBlock

        public PpSymbolBlock​(PaddlePredictor predictor,
                     PpNDManager manager)
        Constructs a new PpSymbolBlock instance.
        Parameters:
        predictor - PaddlePredictor that holds the model information.
        manager - the NDManager to holds the NDArray

    • Method Detail

      • forwardInternal

        protected ai.djl.ndarray.NDList forwardInternal​(ai.djl.training.ParameterStore parameterStore,
                                                ai.djl.ndarray.NDList inputs,
                                                boolean training,
                                                ai.djl.util.PairList<java.lang.String,​java.lang.Object> params)
        Specified by:
        forwardInternal in class ai.djl.nn.AbstractBlock

      • getOutputShapes

        public ai.djl.ndarray.types.Shape[] getOutputShapes​(ai.djl.ndarray.types.Shape[] inputShapes)
        Specified by:
        getOutputShapes in interface ai.djl.nn.Block
        Overrides:
        getOutputShapes in class ai.djl.nn.AbstractSymbolBlock