java.lang.Object
org.apache.wayang.basic.model.op.Op
org.apache.wayang.basic.model.op.nn.Conv
All Implemented Interfaces:
Serializable
Direct Known Subclasses:
Conv2D, Conv3D

public abstract class Conv extends Op
See Also:
  • Field Details

    • inChannels

      protected final int inChannels
    • outChannels

      protected final int outChannels
    • kernelSize

      protected final int[] kernelSize
    • stride

      protected final int[] stride
    • padding

      protected final String padding
    • bias

      protected final boolean bias
  • Constructor Details

    • Conv

      public Conv(int inChannels, int outChannels, int[] kernelSize, int[] stride, String padding, boolean bias, String name, Op.DType dType)
  • Method Details

    • getInChannels

      public int getInChannels()
    • getOutChannels

      public int getOutChannels()
    • getKernelSize

      public int[] getKernelSize()
    • getStride

      public int[] getStride()
    • getPadding

      public String getPadding()
    • getBias

      public boolean getBias()
    • inputsRequired

      public int inputsRequired()
      Specified by:
      inputsRequired in class Op