case class Conv2D(scope: Scope, input: Variable, weight: Variable, bias: Variable, stride: Long, padding: Long, dilation: Long, groups: Long) extends Op with Product with Serializable
2D convolution
- input
batch x in_channels x height x width
- weight
out_channels x in_channels x kernel_size x kernel_size
- bias
out_channels
- returns
Variable with Tensor of size batch x out_channels x L' (length depends on stride/padding/dilation)
- Alphabetic
- By Inheritance
- Conv2D
- Serializable
- Serializable
- Product
- Equals
- Op
- AnyRef
- Any
- Hide All
- Show All
- Public
- All
Instance Constructors
Value Members
-
final
def
!=(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
##(): Int
- Definition Classes
- AnyRef → Any
-
final
def
==(arg0: Any): Boolean
- Definition Classes
- AnyRef → Any
-
final
def
asInstanceOf[T0]: T0
- Definition Classes
- Any
- val batchSize: Long
- val bias: Variable
-
def
clone(): AnyRef
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
- val dilation: Long
-
final
def
eq(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
def
finalize(): Unit
- Attributes
- protected[lang]
- Definition Classes
- AnyRef
- Annotations
- @throws( classOf[java.lang.Throwable] )
-
final
def
getClass(): Class[_]
- Definition Classes
- AnyRef → Any
- Annotations
- @native()
- val groups: Long
- val imageHeight: Long
- val imageWidth: Long
- val input: Variable
- val inputChannels: Long
-
final
def
isInstanceOf[T0]: Boolean
- Definition Classes
- Any
- val kernelSize: Long
-
final
def
ne(arg0: AnyRef): Boolean
- Definition Classes
- AnyRef
-
final
def
notify(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
-
final
def
notifyAll(): Unit
- Definition Classes
- AnyRef
- Annotations
- @native()
- val outChannels: Long
- val padding: Long
-
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
- Conv2D → Op
- See also
The documentation on the trait lamp.autograd.Op for more details and example.
- val scope: Scope
- val stride: Long
-
final
def
synchronized[T0](arg0: ⇒ T0): T0
- Definition Classes
- AnyRef
-
val
value: Variable
The value of this operation
-
final
def
wait(): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long, arg1: Int): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... )
-
final
def
wait(arg0: Long): Unit
- Definition Classes
- AnyRef
- Annotations
- @throws( ... ) @native()
- val weight: Variable