Pooling

Arbitrary-rank pooling configuration.

structure TorchLean.nn.Pooling.Config (d : ℕ) : Type

Configuration shared by arbitrary-dimensional pooling layers.

• kernelSize : Tensor ℕ [d]

  Window extent along each spatial axis.

• stride : Tensor ℕ [d]

  Step along each spatial axis.

• padding : Tensor ℕ [d]

  Symmetric padding along each spatial axis.

def TorchLean.nn.Pooling.Config.outputSpatial {d : ℕ} (config : Config d) (input : Tensor ℕ [d]) : Tensor ℕ [d]

Output grid produced from an input grid by this pooling configuration.

def TorchLean.nn.Pooling.Config.validate {d : ℕ} (config : Config d) (channels : ℕ) (input : Tensor ℕ [d]) (kind : String := "Pooling") : Except String Unit

Validate channel width and pooling geometry.