Package smile.deep.layer

Class LayerBlock

java.lang.Object
smile.deep.layer.LayerBlock
All Implemented Interfaces:
Function<Tensor,Tensor>, Layer
Direct Known Subclasses:
EfficientNet, FusedMBConv, MBConv, SequentialBlock, SqueezeExcitation
public abstract class LayerBlock extends Object implements Layer
A block is combinations of one or more layers. Blocks form the basis of more complex network designs. LayerBlock allows treating the whole container as a single layer, such that performing a transformation on the LayerBlock applies to each of the layers it contains (which are each a registered submodule of the block).

  • Constructor Details

    • LayerBlock

      public LayerBlock()
      Constructor.

    • LayerBlock

      public LayerBlock(String name)
      Constructor.
      Parameters:
      name - the module name.

    • LayerBlock

      public LayerBlock(org.bytedeco.pytorch.Module module)
      Constructor.
      Parameters:
      module - a module.

  • Method Details

    • asTorch

      public org.bytedeco.pytorch.Module asTorch()
      Description copied from interface: Layer
      Returns the PyTorch Module object.
      Specified by:
      asTorch in interface Layer
      Returns:
      the PyTorch Module object.

    • toString

      public String toString()
      Overrides:
      toString in class Object

    • add

      public LayerBlock add(String name, Layer layer)
      Adds a sub-layer.
      Parameters:
      name - the name of sub-layer.
      layer - the sub-layer.
      Returns:
      this object.

    • add

      public LayerBlock add(String name, org.bytedeco.pytorch.Module layer)
      Adds a sub-layer.
      Parameters:
      name - the name of sub-layer.
      layer - the sub-layer.
      Returns:
      this object.