com.intel.analytics.bigdl.nn

SpatialConvolution

class SpatialConvolution[T] extends TensorModule[T] with Initializable

Applies a 2D convolution over an input image composed of several input planes. The input tensor in forward(input) is expected to be a 3D tensor (nInputPlane x height x width).

Annotations
@SerialVersionUID( 8446523046224797382L )
Linear Supertypes
Initializable, TensorModule[T], AbstractModule[Tensor[T], Tensor[T], T], Serializable, Serializable, AnyRef, Any
Known Subclasses
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. SpatialConvolution
  2. Initializable
  3. TensorModule
  4. AbstractModule
  5. Serializable
  6. Serializable
  7. AnyRef
  8. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new SpatialConvolution(nInputPlane: Int, nOutputPlane: Int, kernelW: Int, kernelH: Int, strideW: Int = 1, strideH: Int = 1, padW: Int = 0, padH: Int = 0, nGroup: Int = 1, propagateBack: Boolean = true, wRegularizer: Regularizer[T] = null, bRegularizer: Regularizer[T] = null, initWeight: Tensor[T] = null, initBias: Tensor[T] = null, initGradWeight: Tensor[T] = null, initGradBias: Tensor[T] = null, withBias: Boolean = true)(implicit arg0: ClassTag[T], ev: TensorNumeric[T])

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. val _1x1: Boolean

    Attributes
    protected
  7. def accGradParameters(input: Tensor[T], gradOutput: Tensor[T]): Unit

    Computing the gradient of the module with respect to its own parameters.

    Computing the gradient of the module with respect to its own parameters. Many modules do not perform this step as they do not have any parameters. The state variable name for the parameters is module dependent. The module is expected to accumulate the gradients with respect to the parameters in some variable.

    input
    gradOutput

    Definition Classes
    SpatialConvolutionAbstractModule
  8. def accGradParametersFrame(gradOutput: Tensor[T], gradWeight: Tensor[T], gradBias: Tensor[T], fInput: Tensor[T], scaleW: T, scaleB: T)(implicit ev: TensorNumeric[T]): Unit

    Attributes
    protected
  9. def apply(name: String): Option[AbstractModule[Activity, Activity, T]]

    Find a module with given name.

    Find a module with given name. If there is no module with given name, it will return None. If there are multiple modules with the given name, an exception will be thrown.

    name
    returns

    Definition Classes
    AbstractModule
  10. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  11. var bRegularizer: Regularizer[T]

  12. def backward(input: Tensor[T], gradOutput: Tensor[T]): Tensor[T]

    Performs a back-propagation step through the module, with respect to the given input.

    Performs a back-propagation step through the module, with respect to the given input. In general this method makes the assumption forward(input) has been called before, with the same input. This is necessary for optimization reasons. If you do not respect this rule, backward() will compute incorrect gradients.

    input

    input data

    gradOutput

    gradient of next layer

    returns

    gradient corresponding to input data

    Definition Classes
    AbstractModule
  13. var backwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  14. val bias: Tensor[T]

  15. var biasInitMethod: InitializationMethod

    Attributes
    protected
    Definition Classes
    Initializable
  16. def calcGradParametersFrame(gradOutput: Tensor[T], gradWeight: Tensor[T], gradBias: Tensor[T], fInput: Tensor[T], scaleW: T, scaleB: T)(implicit ev: TensorNumeric[T]): Unit

    Attributes
    protected
  17. def canEqual(other: Any): Boolean

    Definition Classes
    AbstractModule
  18. def checkEngineType(): SpatialConvolution.this.type

    get execution engine type

    get execution engine type

    Definition Classes
    AbstractModule
  19. def clearState(): SpatialConvolution.this.type

    Clear cached activities to save storage space or network bandwidth.

    Clear cached activities to save storage space or network bandwidth. Note that we use Tensor.set to keep some information like tensor share

    The subclass should override this method if it allocate some extra resource, and call the super.clearState in the override method

    returns

    Definition Classes
    SpatialConvolutionAbstractModule
  20. def clone(): AnyRef

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

    Definition Classes
    AbstractModule
  22. var col2imTime: Long

    Attributes
    protected
  23. def copyStatus(src: Module[T]): SpatialConvolution.this.type

    Copy the useful running status from src to this.

    Copy the useful running status from src to this.

    The subclass should override this method if it has some parameters besides weight and bias. Such as runningMean and runningVar of BatchNormalization.

    src

    source Module

    returns

    this

    Definition Classes
    AbstractModule
  24. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  25. def equals(obj: Any): Boolean

    Definition Classes
    SpatialConvolutionAbstractModule → AnyRef → Any
  26. def evaluate(dataSet: LocalDataSet[MiniBatch[T]], vMethods: Array[ValidationMethod[T]]): Array[(ValidationResult, ValidationMethod[T])]

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

    use ValidationMethod to evaluate module

    use ValidationMethod to evaluate module

    dataset

    dataset for test

    vMethods

    validation methods

    batchSize

    total batchsize of all partitions, optional param and default 4 * partitionNum of dataset

    returns

    Definition Classes
    AbstractModule
  28. def evaluate(): SpatialConvolution.this.type

    Definition Classes
    AbstractModule
  29. var fGradInput: Tensor[T]

  30. var fInput: Tensor[T]

  31. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  32. final def forward(input: Tensor[T]): Tensor[T]

    Takes an input object, and computes the corresponding output of the module.

    Takes an input object, and computes the corresponding output of the module. After a forward, the output state variable should have been updated to the new value.

    input

    input data

    returns

    output data

    Definition Classes
    AbstractModule
  33. var forwardTime: Long

    Attributes
    protected
    Definition Classes
    AbstractModule
  34. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  35. def getCol2ImgTime(): Double

  36. def getIm2ColTime(): Double

  37. def getName(): String

    Get the module name, default name is className@namePostfix

    Get the module name, default name is className@namePostfix

    returns

    Definition Classes
    AbstractModule
  38. def getNumericType(): TensorDataType

    returns

    Float or Double

    Definition Classes
    AbstractModule
  39. def getParameters(): (Tensor[T], Tensor[T])

    This method compact all parameters and gradients of the model into two tensors.

    This method compact all parameters and gradients of the model into two tensors. So it's easier to use optim method

    returns

    Definition Classes
    AbstractModule
  40. def getParametersTable(): Table

    This function returns a table contains ModuleName, the parameter names and parameter value in this module.

    This function returns a table contains ModuleName, the parameter names and parameter value in this module. The result table is a structure of Table(ModuleName -> Table(ParameterName -> ParameterValue)), and the type is Table[String, Table[String, Tensor[T]]].

    For example, get the weight of a module named conv1: table[Table]("conv1")[Tensor[T]]("weight").

    Custom modules should override this function if they have parameters.

    returns

    Table

    Definition Classes
    SpatialConvolutionAbstractModule
  41. def getPrintName(): String

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

    Get the scale of gradientBias

    Get the scale of gradientBias

    Definition Classes
    AbstractModule
  43. def getScaleW(): Double

    Get the scale of gradientWeight

    Get the scale of gradientWeight

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

    Definition Classes
    AbstractModule
  45. def getWeightsBias(): Array[Tensor[T]]

    Get weight and bias for the module

    Get weight and bias for the module

    returns

    array of weights and bias

    Definition Classes
    AbstractModule
  46. val gradBias: Tensor[T]

  47. var gradInput: Tensor[T]

    The cached gradient of activities.

    The cached gradient of activities. So we don't compute it again when need it

    Definition Classes
    AbstractModule
  48. val gradWeight: Tensor[T]

  49. var gradWeightMM: Tensor[T]

    Attributes
    protected
  50. var gradWeightMMInBatch: Tensor[T]

    Attributes
    protected
  51. val gradientBiasMT: Tensor[T]

    Attributes
    protected
  52. def hashCode(): Int

    Definition Classes
    SpatialConvolutionAbstractModule → AnyRef → Any
  53. var im2colTime: Long

    Attributes
    protected
  54. val initBias: Tensor[T]

  55. val initGradBias: Tensor[T]

  56. val initGradWeight: Tensor[T]

  57. val initWeight: Tensor[T]

  58. def inputs(nodes: ModuleNode[T]*): ModuleNode[T]

    Some other modules point to current module

    Some other modules point to current module

    nodes

    upstream module nodes

    returns

    node containing current module

    Definition Classes
    AbstractModule
  59. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  60. final def isTraining(): Boolean

    Definition Classes
    AbstractModule
  61. val kernelH: Int

  62. val kernelW: Int

  63. var line: String

    Attributes
    protected
    Definition Classes
    AbstractModule
  64. def loadModelWeights(srcModel: Module[Float], matchAll: Boolean = true): SpatialConvolution.this.type

    copy weights from another model, mapping by layer name

    copy weights from another model, mapping by layer name

    srcModel

    model to copy from

    matchAll

    whether to match all layers' weights and bias,

    returns

    current module

    Definition Classes
    AbstractModule
  65. def loadWeights(weightPath: String, matchAll: Boolean = true): SpatialConvolution.this.type

    load pretrained weights and bias to current module

    load pretrained weights and bias to current module

    weightPath

    file to store weights and bias

    matchAll

    whether to match all layers' weights and bias, if not, only load existing pretrained weights and bias

    returns

    current module

    Definition Classes
    AbstractModule
  66. val nGroup: Int

  67. val nInputPlane: Int

  68. val nOutputPlane: Int

  69. final def ne(arg0: AnyRef): Boolean

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

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

    Definition Classes
    AnyRef
  72. val ones: Tensor[T]

    Attributes
    protected
  73. val onesBatch: Tensor[T]

    Attributes
    protected
  74. val onesBias: Tensor[T]

    Attributes
    protected
  75. var output: Tensor[T]

    The cached output.

    The cached output. So we don't compute it again when need it

    Definition Classes
    AbstractModule
  76. val padH: Int

  77. val padW: Int

  78. def parameters(): (Array[Tensor[T]], Array[Tensor[T]])

    This function returns two arrays.

    This function returns two arrays. One for the weights and the other the gradients Custom modules should override this function if they have parameters

    returns

    (Array of weights, Array of grad)

    Definition Classes
    SpatialConvolutionAbstractModule
  79. def predict(dataset: RDD[Sample[T]]): RDD[Activity]

    module predict, return the probability distribution

    module predict, return the probability distribution

    dataset

    dataset for prediction

    Definition Classes
    AbstractModule
  80. def predictClass(dataset: RDD[Sample[T]]): RDD[Int]

    module predict, return the predict label

    module predict, return the predict label

    dataset

    dataset for prediction

    Definition Classes
    AbstractModule
  81. val propagateBack: Boolean

  82. def reset(): Unit

  83. def resetTimes(): Unit

    Definition Classes
    AbstractModule
  84. var results: Array[Future[Unit]]

    Attributes
    protected
  85. def save(path: String, overWrite: Boolean = false): SpatialConvolution.this.type

    Save this module to path.

    Save this module to path.

    path

    path to save module, local file system, HDFS and Amazon S3 is supported. HDFS path should be like "hdfs://[host]:[port]/xxx" Amazon S3 path should be like "s3a://bucket/xxx"

    overWrite

    if overwrite

    returns

    self

    Definition Classes
    AbstractModule
  86. def saveCaffe(prototxtPath: String, modelPath: String, useV2: Boolean = true, overwrite: Boolean = false): SpatialConvolution.this.type

    Definition Classes
    AbstractModule
  87. def saveTF(inputs: Seq[(String, Seq[Int])], path: String, byteOrder: ByteOrder = ByteOrder.LITTLE_ENDIAN, dataFormat: TensorflowDataFormat = TensorflowDataFormat.NHWC): SpatialConvolution.this.type

    Definition Classes
    AbstractModule
  88. def saveTorch(path: String, overWrite: Boolean = false): SpatialConvolution.this.type

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

    save weights and bias to file

    save weights and bias to file

    path

    file to save

    overWrite

    whether to overwrite or not

    Definition Classes
    AbstractModule
  90. var scaleB: Double

    Attributes
    protected
    Definition Classes
    AbstractModule
  91. var scaleW: Double

    The scale of gradient weight and gradient bias before gradParameters being accumulated.

    The scale of gradient weight and gradient bias before gradParameters being accumulated.

    Attributes
    protected
    Definition Classes
    AbstractModule
  92. def setInitMethod(weightInitMethod: InitializationMethod = null, biasInitMethod: InitializationMethod = null): SpatialConvolution.this.type

    Definition Classes
    Initializable
  93. def setLine(line: String): SpatialConvolution.this.type

    Definition Classes
    AbstractModule
  94. def setName(name: String): SpatialConvolution.this.type

    Set the module name

    Set the module name

    name
    returns

    Definition Classes
    AbstractModule
  95. def setScaleB(b: Double): SpatialConvolution.this.type

    Set the scale of gradientBias

    Set the scale of gradientBias

    b

    the value of the scale of gradientBias

    returns

    this

    Definition Classes
    AbstractModule
  96. def setScaleW(w: Double): SpatialConvolution.this.type

    Set the scale of gradientWeight

    Set the scale of gradientWeight

    w

    the value of the scale of gradientWeight

    returns

    this

    Definition Classes
    AbstractModule
  97. def setWeightsBias(newWeights: Array[Tensor[T]]): SpatialConvolution.this.type

    Set weight and bias for the module

    Set weight and bias for the module

    newWeights

    array of weights and bias

    returns

    Definition Classes
    AbstractModule
  98. val strideH: Int

  99. val strideW: Int

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

    Definition Classes
    AnyRef
  101. def toString(): String

    Definition Classes
    SpatialConvolutionAbstractModule → AnyRef → Any
  102. var train: Boolean

    Module status.

    Module status. It is useful for modules like dropout/batch normalization

    Attributes
    protected
    Definition Classes
    AbstractModule
  103. def training(): SpatialConvolution.this.type

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

    Computing the gradient of the module with respect to its own input.

    Computing the gradient of the module with respect to its own input. This is returned in gradInput. Also, the gradInput state variable is updated accordingly.

    input
    gradOutput
    returns

    Definition Classes
    SpatialConvolutionAbstractModule
  105. def updateGradInputFrame(gradInput: Tensor[T], gradOutput: Tensor[T], weight: Tensor[T], fgradInput: Tensor[T], kW: Int, kH: Int, dW: Int, dH: Int, padW: Int, padH: Int)(implicit ev: TensorNumeric[T]): Unit

    Attributes
    protected
  106. def updateOutput(input: Tensor[T]): Tensor[T]

    Computes the output using the current parameter set of the class and input.

    Computes the output using the current parameter set of the class and input. This function returns the result which is stored in the output field.

    input
    returns

    Definition Classes
    SpatialConvolutionAbstractModule
  107. def updateOutputFrame(input: Tensor[T], output: Tensor[T], weight: Tensor[T], bias: Tensor[T], fInput: Tensor[T], kW: Int, kH: Int, dW: Int, dH: Int, padW: Int, padH: Int, nInputPlane: Int, inputWidth: Int, inputHeight: Int, nOutputPlane: Int, outputWidth: Int, outputHeight: Int)(implicit ev: TensorNumeric[T]): Unit

    Attributes
    protected
  108. def updateParameters(learningRate: T): Unit

    Definition Classes
    SpatialConvolutionAbstractModule
  109. var wRegularizer: Regularizer[T]

  110. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  113. val weight: Tensor[T]

  114. var weightInitMethod: InitializationMethod

    Attributes
    protected
    Definition Classes
    Initializable
  115. var weightMM: Tensor[T]

    Attributes
    protected
  116. val withBias: Boolean

  117. def zeroGradParameters(): Unit

    If the module has parameters, this will zero the accumulation of the gradients with respect to these parameters.

    If the module has parameters, this will zero the accumulation of the gradients with respect to these parameters. Otherwise, it does nothing.

    Definition Classes
    SpatialConvolutionAbstractModule

Inherited from Initializable

Inherited from TensorModule[T]

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

Inherited from Serializable

Inherited from Serializable

Inherited from AnyRef

Inherited from Any

Ungrouped