org.apache.mxnet.javaapi

BatchNormParam

Related Doc: package javaapi

class BatchNormParam extends AnyRef

This Param Object is specifically used for BatchNorm

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

Instance Constructors

  1. new BatchNormParam(data: NDArray, gamma: NDArray, beta: NDArray, moving_mean: NDArray, moving_var: NDArray)

    data

    Input data to batch normalization

    gamma

    gamma array

    beta

    beta array

    moving_mean

    running mean of input

    moving_var

    running variance of input

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 getAxis(): Integer

  10. def getBeta(): NDArray

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

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

  13. def getData(): NDArray

  14. def getEps(): Double

  15. def getFix_gamma(): Boolean

  16. def getGamma(): NDArray

  17. def getMax_calib_range(): Float

  18. def getMin_calib_range(): Float

  19. def getMomentum(): Float

  20. def getMoving_mean(): NDArray

  21. def getMoving_var(): NDArray

  22. def getOut(): mxnet.NDArray

  23. def getOutput_mean_var(): Boolean

  24. def getUse_global_stats(): Boolean

  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 setAxis(axis: Integer): BatchNormParam

    axis

    Specify which shape axis the channel is specified

  31. def setCudnn_off(cudnn_off: Boolean): BatchNormParam

    cudnn_off

    Do not select CUDNN operator, if available

  32. def setEps(eps: Double): BatchNormParam

    eps

    Epsilon to prevent div 0. Must be no less than CUDNN_BN_MIN_EPSILON defined in cudnn.h when using cudnn (usually 1e-5)

  33. def setFix_gamma(fix_gamma: Boolean): BatchNormParam

    fix_gamma

    Fix gamma while training

  34. def setMax_calib_range(max_calib_range: Float): BatchNormParam

    max_calib_range

    The maximum scalar value in the form of float32 obtained through calibration. If present, it will be used to by quantized batch norm op to calculate primitive scale.Note: this calib_range is to calib bn output.

  35. def setMin_calib_range(min_calib_range: Float): BatchNormParam

    min_calib_range

    The minimum scalar value in the form of float32 obtained through calibration. If present, it will be used to by quantized batch norm op to calculate primitive scale.Note: this calib_range is to calib bn output.

  36. def setMomentum(momentum: Float): BatchNormParam

    momentum

    Momentum for moving average

  37. def setOut(out: NDArray): BatchNormParam

  38. def setOutput_mean_var(output_mean_var: Boolean): BatchNormParam

    output_mean_var

    Output the mean and inverse std

  39. def setUse_global_stats(use_global_stats: Boolean): BatchNormParam

    use_global_stats

    Whether use global moving statistics instead of local batch-norm. This will force change batch-norm into a scale shift operator.

  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