Convolution
case class Convolution(scope: Scope, input: Variable, weight: Variable, bias: Variable, stride: Array[Long], padding: Array[Long], dilation: Array[Long], transposed: Boolean, outputPadding: Array[Long], groups: Long) extends Op
1D/2D/3D convolution
- Value parameters:
- bias
out_channels
- input
batch x in_channels x height x width
- weight
out_channels x in_channels x kernel_size x kernel_size
- Returns:
Variable with Tensor of size batch x out_channels x L' (length depends on stride/padding/dilation)