Packages

c

lamp.autograd

BatchNorm2D

case class BatchNorm2D(scope: Scope, input: Variable, weight: Variable, bias: Variable, runningMean: STen, runningVar: STen, training: Boolean, momentum: Double, eps: Double) extends Op with Product with Serializable

Batch Norm 2D 0-th dimension are samples. 1-th are features, everything else is averaged out.

Linear Supertypes
Serializable, Product, Equals, Op, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. BatchNorm2D
  2. Serializable
  3. Product
  4. Equals
  5. Op
  6. AnyRef
  7. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Instance Constructors

  1. new BatchNorm2D(scope: Scope, input: Variable, weight: Variable, bias: Variable, runningMean: STen, runningVar: STen, training: Boolean, momentum: Double, eps: Double)

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. val bias: Variable
  6. def clone(): AnyRef
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.CloneNotSupportedException]) @native() @IntrinsicCandidate()
  7. val eps: Double
  8. final def eq(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  9. val expectedShape: List[Long]
  10. final def getClass(): Class[_ <: AnyRef]
    Definition Classes
    AnyRef → Any
    Annotations
    @native() @IntrinsicCandidate()
  11. val input: Variable
  12. val inputShape: List[Long]
  13. final def isInstanceOf[T0]: Boolean
    Definition Classes
    Any
  14. val momentum: Double
  15. final def ne(arg0: AnyRef): Boolean
    Definition Classes
    AnyRef
  16. final def notify(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @IntrinsicCandidate()
  17. final def notifyAll(): Unit
    Definition Classes
    AnyRef
    Annotations
    @native() @IntrinsicCandidate()
  18. val output: Tensor
  19. val params: List[(Variable, (STen, STen) => Unit)]

    Implementation of the backward pass

    Implementation of the backward pass

    A list of input variables paired up with an anonymous function computing the respective partial derivative. With the notation in the documentation of the trait lamp.autograd.Op: dy/dw2 => dy/dw2 * dw2/dw1. The first argument of the anonymous function is the incoming partial derivative (dy/dw2), the second argument is the output tensor into which the result (dy/dw2 * dw2/dw1) is accumulated (added).

    If the operation does not support computing the partial derivative for some of its arguments, then do not include that argument in this list.

    Definition Classes
    BatchNorm2DOp
    See also

    The documentation on the trait lamp.autograd.Op for more details and example.

  20. def productElementNames: Iterator[String]
    Definition Classes
    Product
  21. val runningMean: STen
  22. val runningVar: STen
  23. val saveInvstd: Tensor
  24. val saveMean: Tensor
  25. val scope: Scope
  26. final def synchronized[T0](arg0: => T0): T0
    Definition Classes
    AnyRef
  27. val training: Boolean
  28. val value: Variable

    The value of this operation

    The value of this operation

    Definition Classes
    BatchNorm2DOp
  29. final def wait(arg0: Long, arg1: Int): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  30. final def wait(arg0: Long): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException]) @native()
  31. final def wait(): Unit
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.InterruptedException])
  32. val weight: Variable

Deprecated Value Members

  1. def finalize(): Unit
    Attributes
    protected[lang]
    Definition Classes
    AnyRef
    Annotations
    @throws(classOf[java.lang.Throwable]) @Deprecated
    Deprecated

Inherited from Serializable

Inherited from Product

Inherited from Equals

Inherited from Op

Inherited from AnyRef

Inherited from Any

Ungrouped