com.intel.analytics.zoo.pipeline.api.keras.layers

BatchNormalization

class BatchNormalization[T] extends bigdl.nn.keras.BatchNormalization[T] with Net

Batch normalization layer. Normalize the activations of the previous layer at each batch, i.e. applies a transformation that maintains the mean activation close to 0 and the activation standard deviation close to 1. It is a feature-wise normalization, each feature map in the input will be normalized separately. The input of this layer should be 4D.

When you use this layer as the first layer of a model, you need to provide the argument inputShape (a Single Shape, does not include the batch dimension).

T

Numeric type of parameter(e.g. weight, bias). Only support float/double now.

Linear Supertypes
Net, bigdl.nn.keras.BatchNormalization[T], KerasLayer[Tensor[T], Tensor[T], T], Container[Tensor[T], Tensor[T], T], AbstractModule[Tensor[T], Tensor[T], T], InferShape, Serializable, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. BatchNormalization
  2. Net
  3. BatchNormalization
  4. KerasLayer
  5. Container
  6. AbstractModule
  7. InferShape
  8. Serializable
  9. Serializable
  10. AnyRef
  11. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new BatchNormalization(epsilon: Double = 0.001, momentum: Double = 0.99, betaInit: String = "zero", gammaInit: String = "one", dimOrdering: DataFormat = ..., inputShape: Shape = null)(implicit arg0: ClassTag[T], ev: TensorNumeric[T])

    epsilon

    Small Double > 0. Fuzz parameter. Default is 0.001.

    momentum

    Double. Momentum in the computation of the exponential average of the mean and standard deviation of the data, for feature-wise normalization. Default is 0.99.

    betaInit

    Name of initialization function for shift parameter. Default is 'zero'.

    gammaInit

    Name of initialization function for scale parameter. Default is 'one'.

    dimOrdering

    Format of input data. Either DataFormat.NCHW (dimOrdering='th') or DataFormat.NHWC (dimOrdering='tf'). Default is NCHW. For NCHW, axis along which to normalize is 1. For NHWC, axis is 3.

    inputShape

    A Single Shape, does not include the batch dimension.

Value Members

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

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

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

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

    Definition Classes
    Any
  6. def accGradParameters(input: Tensor[T], gradOutput: Tensor[T]): Unit

    Definition Classes
    KerasLayer → AbstractModule
  7. def apply(name: String): Option[AbstractModule[Activity, Activity, T]]

    Definition Classes
    Container → AbstractModule
  8. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  9. def backward(input: Tensor[T], gradOutput: Tensor[T]): Tensor[T]

    Definition Classes
    AbstractModule
  10. var backwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  11. val betaInit: String

    Name of initialization function for shift parameter.

    Name of initialization function for shift parameter. Default is 'zero'.

    Definition Classes
    BatchNormalization → BatchNormalization
  12. def build(calcInputShape: Shape): Shape

    Definition Classes
    KerasLayer → InferShape
  13. def canEqual(other: Any): Boolean

    Definition Classes
    Container → AbstractModule
  14. final def checkEngineType(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  15. def clearState(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  16. final def clone(deepCopy: Boolean): AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    AbstractModule
  17. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  18. final def cloneModule(): AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    AbstractModule
  19. def computeOutputShape(inputShape: Shape): Shape

    Definition Classes
    BatchNormalization → KerasLayer → InferShape
  20. val dimOrdering: DataFormat

    Format of input data.

    Format of input data. Either DataFormat.NCHW (dimOrdering='th') or DataFormat.NHWC (dimOrdering='tf'). Default is NCHW. For NCHW, axis along which to normalize is 1. For NHWC, axis is 3.

    Definition Classes
    BatchNormalization → BatchNormalization
  21. def doBuild(inputShape: Shape): AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    BatchNormalization → KerasLayer
  22. val epsilon: Double

    Small Double > 0.

    Small Double > 0. Fuzz parameter. Default is 0.001.

    Definition Classes
    BatchNormalization → BatchNormalization
  23. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  24. def equals(other: Any): Boolean

    Definition Classes
    Container → AbstractModule → AnyRef → Any
  25. final def evaluate(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  26. final def evaluate(dataSet: LocalDataSet[MiniBatch[T]], vMethods: Array[_ <: ValidationMethod[T]]): Array[(ValidationResult, ValidationMethod[T])]

    Definition Classes
    AbstractModule
  27. final def evaluate(dataset: RDD[Sample[T]], vMethods: Array[_ <: ValidationMethod[T]], batchSize: Option[Int]): Array[(ValidationResult, ValidationMethod[T])]

    Definition Classes
    AbstractModule
  28. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  29. def findModules(moduleType: String): ArrayBuffer[AbstractModule[_, _, T]]

    Definition Classes
    Container
  30. final def forward(input: Tensor[T]): Tensor[T]

    Definition Classes
    AbstractModule
  31. var forwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  32. def freeze(names: String*): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  33. def from[T](vars: Variable[T]*)(implicit arg0: ClassTag[T], ev: TensorNumeric[T]): Variable[T]

    Build graph: some other modules point to current module

    Build graph: some other modules point to current module

    vars

    upstream variables

    returns

    Variable containing current module

    Definition Classes
    Net
  34. val gammaInit: String

    Name of initialization function for scale parameter.

    Name of initialization function for scale parameter. Default is 'one'.

    Definition Classes
    BatchNormalization → BatchNormalization
  35. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  36. def getExtraParameter(): Array[Tensor[T]]

    Definition Classes
    Container → AbstractModule
  37. final def getInputShape(): Shape

    Definition Classes
    InferShape
  38. final def getName(): String

    Definition Classes
    AbstractModule
  39. final def getNumericType(): TensorDataType

    Definition Classes
    AbstractModule
  40. final def getOutputShape(): Shape

    Definition Classes
    InferShape
  41. def getParametersTable(): Table

    Definition Classes
    Container → AbstractModule
  42. final def getPrintName(): String

    Attributes
    protected
    Definition Classes
    AbstractModule
  43. final def getScaleB(): Double

    Definition Classes
    AbstractModule
  44. final def getScaleW(): Double

    Definition Classes
    AbstractModule
  45. def getTimes(): Array[(AbstractModule[_ <: Activity, _ <: Activity, T], Long, Long)]

    Definition Classes
    Container → AbstractModule
  46. final def getWeightsBias(): Array[Tensor[T]]

    Definition Classes
    AbstractModule
  47. var gradInput: Tensor[T]

    Definition Classes
    AbstractModule
  48. final def hasName: Boolean

    Definition Classes
    AbstractModule
  49. def hashCode(): Int

    Definition Classes
    Container → AbstractModule → AnyRef → Any
  50. val inputShape: Shape

    A Single Shape, does not include the batch dimension.

    A Single Shape, does not include the batch dimension.

    Definition Classes
    BatchNormalization → BatchNormalization
  51. def inputs(first: (ModuleNode[T], Int), nodesWithIndex: (ModuleNode[T], Int)*): ModuleNode[T]

    Definition Classes
    KerasLayer → AbstractModule
  52. def inputs(nodes: Array[ModuleNode[T]]): ModuleNode[T]

    Definition Classes
    KerasLayer → AbstractModule
  53. def inputs(nodes: ModuleNode[T]*): ModuleNode[T]

    Definition Classes
    KerasLayer → AbstractModule
  54. def isBuilt(): Boolean

    Definition Classes
    KerasLayer → InferShape
  55. def isFrozen[T]()(implicit arg0: ClassTag[T]): Boolean

    Definition Classes
    Net
  56. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  57. def isKerasStyle(): Boolean

    Definition Classes
    KerasLayer → InferShape
  58. final def isTraining(): Boolean

    Definition Classes
    AbstractModule
  59. def labor: AbstractModule[Tensor[T], Tensor[T], T]

    Definition Classes
    KerasLayer
  60. def labor_=(value: AbstractModule[Tensor[T], Tensor[T], T]): Unit

    Definition Classes
    KerasLayer
  61. var line: String

    Attributes
    protected
    Definition Classes
    AbstractModule
  62. final def loadModelWeights(srcModel: Module[Float], matchAll: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  63. final def loadWeights(weightPath: String, matchAll: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  64. val modules: ArrayBuffer[AbstractModule[Activity, Activity, T]]

    Definition Classes
    Container
  65. val momentum: Double

    Double.

    Double. Momentum in the computation of the exponential average of the mean and standard deviation of the data, for feature-wise normalization. Default is 0.99.

    Definition Classes
    BatchNormalization → BatchNormalization
  66. final def ne(arg0: AnyRef): Boolean

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

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

    Definition Classes
    AnyRef
  69. var output: Tensor[T]

    Definition Classes
    AbstractModule
  70. def parameters(): (Array[Tensor[T]], Array[Tensor[T]])

    Definition Classes
    Container → AbstractModule
  71. final def predict(dataset: RDD[Sample[T]], batchSize: Int, shareBuffer: Boolean): RDD[Activity]

    Definition Classes
    AbstractModule
  72. final def predictClass(dataset: RDD[Sample[T]], batchSize: Int): RDD[Int]

    Definition Classes
    AbstractModule
  73. final def predictImage(imageFrame: ImageFrame, outputLayer: String, shareBuffer: Boolean, batchPerPartition: Int, predictKey: String, featurePaddingParam: Option[PaddingParam[T]]): ImageFrame

    Definition Classes
    AbstractModule
  74. def processInputs(first: (ModuleNode[T], Int), nodesWithIndex: (ModuleNode[T], Int)*): ModuleNode[T]

    Attributes
    protected
    Definition Classes
    AbstractModule
  75. def processInputs(nodes: Seq[ModuleNode[T]]): ModuleNode[T]

    Attributes
    protected
    Definition Classes
    AbstractModule
  76. final def quantize(): Module[T]

    Definition Classes
    AbstractModule
  77. def reset(): Unit

    Definition Classes
    Container → AbstractModule
  78. def resetTimes(): Unit

    Definition Classes
    Container → AbstractModule
  79. final def saveCaffe(prototxtPath: String, modelPath: String, useV2: Boolean, overwrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  80. final def saveDefinition(path: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  81. final def saveModule(path: String, weightPath: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  82. final def saveTF(inputs: Seq[(String, Seq[Int])], path: String, byteOrder: ByteOrder, dataFormat: TensorflowDataFormat): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  83. final def saveTorch(path: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  84. final def saveWeights(path: String, overWrite: Boolean): Unit

    Definition Classes
    AbstractModule
  85. var scaleB: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  86. var scaleW: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  87. final def setExtraParameter(extraParam: Array[Tensor[T]]): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  88. final def setLine(line: String): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  89. final def setName(name: String): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  90. def setScaleB(b: Double): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  91. def setScaleW(w: Double): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  92. final def setWeightsBias(newWeights: Array[Tensor[T]]): BatchNormalization.this.type

    Definition Classes
    AbstractModule
  93. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  94. def toGraph(startNodes: ModuleNode[T]*): Graph[T]

    Definition Classes
    AbstractModule
  95. def toString(): String

    Definition Classes
    AbstractModule → AnyRef → Any
  96. var train: Boolean

    Attributes
    protected
    Definition Classes
    AbstractModule
  97. final def training(): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  98. def unFreeze(names: String*): BatchNormalization.this.type

    Definition Classes
    Container → AbstractModule
  99. def updateGradInput(input: Tensor[T], gradOutput: Tensor[T]): Tensor[T]

    Definition Classes
    KerasLayer → AbstractModule
  100. def updateOutput(input: Tensor[T]): Tensor[T]

    Definition Classes
    KerasLayer → AbstractModule
  101. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  104. final def zeroGradParameters(): Unit

    Definition Classes
    AbstractModule

Deprecated Value Members

  1. final def save(path: String, overWrite: Boolean): BatchNormalization.this.type

    Definition Classes
    AbstractModule
    Annotations
    @deprecated
    Deprecated

    (Since version 0.3.0) please use recommended saveModule(path, overWrite)

Inherited from Net

Inherited from bigdl.nn.keras.BatchNormalization[T]

Inherited from KerasLayer[Tensor[T], Tensor[T], T]

Inherited from Container[Tensor[T], Tensor[T], T]

Inherited from AbstractModule[Tensor[T], Tensor[T], T]

Inherited from InferShape

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped