org.apache.mxnet.javaapi

ConvolutionParam

Related Doc: package javaapi

class ConvolutionParam extends AnyRef

This Param Object is specifically used for Convolution

Linear Supertypes
AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. ConvolutionParam
  2. AnyRef
  3. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new ConvolutionParam(data: NDArray, weight: NDArray, bias: NDArray, kernel: Shape, num_filter: Integer)

    data

    Input data to the ConvolutionOp.

    weight

    Weight matrix.

    bias

    Bias parameter.

    kernel

    Convolution kernel size: (w,), (h, w) or (d, h, w)

    num_filter

    Convolution filter(channel) number

Value Members

  1. final def !=(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  2. final def ##(): Int

    Definition Classes
    AnyRef → Any
  3. final def ==(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  4. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  5. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  6. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  7. def equals(arg0: Any): Boolean

    Definition Classes
    AnyRef → Any
  8. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  9. def getBias(): NDArray

  10. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  11. def getCudnn_off(): Boolean

  12. def getCudnn_tune(): String

  13. def getData(): NDArray

  14. def getDilate(): Shape

  15. def getKernel(): Shape

  16. def getLayout(): String

  17. def getNo_bias(): Boolean

  18. def getNum_filter(): Integer

  19. def getNum_group(): Integer

  20. def getOut(): mxnet.NDArray

  21. def getPad(): Shape

  22. def getStride(): Shape

  23. def getWeight(): NDArray

  24. def getWorkspace(): Long

  25. def hashCode(): Int

    Definition Classes
    AnyRef → Any
  26. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  27. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  28. final def notify(): Unit

    Definition Classes
    AnyRef
  29. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  30. def setCudnn_off(cudnn_off: Boolean): ConvolutionParam

    cudnn_off

    Turn off cudnn for this layer.

  31. def setCudnn_tune(cudnn_tune: String): ConvolutionParam

    cudnn_tune

    Whether to pick convolution algo by running performance test.

  32. def setDilate(dilate: Shape): ConvolutionParam

    dilate

    Convolution dilate: (w,), (h, w) or (d, h, w). Defaults to 1 for each dimension.

  33. def setLayout(layout: String): ConvolutionParam

    layout

    Set layout for input, output and weight. Empty for default layout: NCW for 1d, NCHW for 2d and NCDHW for 3d.NHWC and NDHWC are only supported on GPU.

  34. def setNo_bias(no_bias: Boolean): ConvolutionParam

    no_bias

    Whether to disable bias parameter.

  35. def setNum_group(num_group: Integer): ConvolutionParam

    num_group

    Number of group partitions.

  36. def setOut(out: NDArray): ConvolutionParam

  37. def setPad(pad: Shape): ConvolutionParam

    pad

    Zero pad for convolution: (w,), (h, w) or (d, h, w). Defaults to no padding.

  38. def setStride(stride: Shape): ConvolutionParam

    stride

    Convolution stride: (w,), (h, w) or (d, h, w). Defaults to 1 for each dimension.

  39. def setWorkspace(workspace: Long): ConvolutionParam

    workspace

    Maximum temporary workspace allowed (MB) in convolution.This parameter has two usages. When CUDNN is not used, it determines the effective batch size of the convolution kernel. When CUDNN is used, it controls the maximum temporary storage used for tuning the best CUDNN kernel when limited_workspace strategy is used.

  40. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  41. def toString(): String

    Definition Classes
    AnyRef → Any
  42. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  43. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  44. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from AnyRef

Inherited from Any

Ungrouped