Package org.deeplearning4j.nn.conf.graph

Class ReshapeVertex

    • Field Detail

      • reshapeOrder

        protected char reshapeOrder

      • newShape

        protected int[] newShape

      • maskShape

        protected int[] maskShape

    • Constructor Detail

      • ReshapeVertex

        public ReshapeVertex​(int... newShape)
        Reshape with the default reshape order of 'c'
        Parameters:
        newShape - New shape for activations

      • ReshapeVertex

        public ReshapeVertex​(char reshapeOrder,
                     int[] newShape,
                     int[] maskShape)
        Parameters:
        reshapeOrder - Order (must be 'c' or 'f') for the activations
        newShape - New shape
        maskShape - Mask shape

    • Method Detail

      • numParams

        public long numParams​(boolean backprop)
        Specified by:
        numParams in class GraphVertex

      • minVertexInputs

        public int minVertexInputs()
        Specified by:
        minVertexInputs in class GraphVertex
        Returns:
        The Smallest valid number of inputs to this vertex

      • maxVertexInputs

        public int maxVertexInputs()
        Specified by:
        maxVertexInputs in class GraphVertex
        Returns:
        The largest valid number of inputs to this vertex

      • instantiate

        public GraphVertex instantiate​(ComputationGraph graph,
                               String name,
                               int idx,
                               INDArray paramsView,
                               boolean initializeParams,
                               DataType networkDatatype)
        Description copied from class: GraphVertex
        Create a GraphVertex instance, for the given computation graph, given the configuration instance.
        Specified by:
        instantiate in class GraphVertex
        Parameters:
        graph - The computation graph that this GraphVertex is to be part of
        name - The name of the GraphVertex object
        idx - The index of the GraphVertex
        paramsView - A view of the full parameters array
        initializeParams - If true: initialize the parameters. If false: make no change to the values in the paramsView array
        Returns:
        The implementation GraphVertex object (i.e., implementation, no the configuration)

      • getOutputType

        public InputType getOutputType​(int layerIndex,
                               InputType... vertexInputs)
                        throws InvalidInputTypeException
        Description copied from class: GraphVertex
        Determine the type of output for this GraphVertex, given the specified inputs. Given that a GraphVertex may do arbitrary processing or modifications of the inputs, the output types can be quite different to the input type(s).
        This is generally used to determine when to add preprocessors, as well as the input sizes etc for layers
        Specified by:
        getOutputType in class GraphVertex
        Parameters:
        layerIndex - The index of the layer (if appropriate/necessary).
        vertexInputs - The inputs to this vertex
        Returns:
        The type of output for this vertex
        Throws:
        InvalidInputTypeException - If the input type is invalid for this type of GraphVertex

      • getMemoryReport

        public MemoryReport getMemoryReport​(InputType... inputTypes)
        Description copied from class: GraphVertex
        This is a report of the estimated memory consumption for the given vertex
        Specified by:
        getMemoryReport in class GraphVertex
        Parameters:
        inputTypes - Input types to the vertex. Memory consumption is often a function of the input type
        Returns:
        Memory report for the vertex