lamp-core/lamp.autograd/MaxPool2D

MaxPool2D

case class MaxPool2D(scope: Scope, input: Variable, kernelSize: Long, stride: Long, padding: Long, dilation: Long) extends Op

2D max pooling

Value parameters:
input

batch x in_channels x h x w

trait Serializable
trait Product
trait Equals
trait Op
class Object
trait Matchable
class Any

Value members

Inherited methods

def productElementNames: Iterator[String]
Inherited from:
Product
def productIterator: Iterator[Any]
Inherited from:
Product

Concrete fields

val batchSize: Long
val imageSize: Long
val inputChannels: Long
val mask: Tensor
val output: Tensor
override val params: List[(Variable, (STen, STen) => Unit)]