Class CapsuleStrengthLayer

    • Method Detail

      • defineLayer

        public SDVariable defineLayer​(SameDiff SD,
                                      SDVariable layerInput)
        Description copied from class: SameDiffLambdaLayer
        The defineLayer method is used to define the forward pass for the layer
        Specified by:
        defineLayer in class SameDiffLambdaLayer
        Parameters:
        SD - SameDiff instance to use to define the vertex
        layerInput - Layer input variable
        Returns:
        The output variable (corresponding to the output activations for the layer)
      • getOutputType

        public InputType getOutputType​(int layerIndex,
                                       InputType inputType)
        Description copied from class: Layer
        For a given type of input to this layer, what is the type of the output?
        Overrides:
        getOutputType in class SameDiffLambdaLayer
        Parameters:
        layerIndex - Index of the layer
        inputType - Type of input for the layer
        Returns:
        Type of output from the layer