Class MaxPool2dLayer

java.lang.Object

smile.deep.layer.ModuleLayer

smile.deep.layer.MaxPool2dLayer

All Implemented Interfaces:

Function<Tensor,Tensor>, Layer

public class MaxPool2dLayer
extends ModuleLayer

A max pooling layer that reduces a tensor by combining cells, and assigning the maximum value of the input cells to the output cell.

Constructor Summary

Constructors

Constructor	Description
MaxPool2dLayer(int kernel)	Constructor.
MaxPool2dLayer(int height, int width)	Constructor.

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor input)	Forward propagation (or forward pass) through the layer.

Methods inherited from class ModuleLayer

module, toString

Methods inherited from class Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Methods inherited from interface Function

andThen, compose

Methods inherited from interface Layer

apply, name, to, to

Constructor Details

MaxPool2dLayer

public MaxPool2dLayer(int kernel)

Constructor.

Parameters:

kernel - the window/kernel size.

MaxPool2dLayer

public MaxPool2dLayer(int height,
 int width)

Constructor.

Parameters:

height - the window/kernel height.

width - the window/kernel width.

Method Details

forward

public Tensor forward(Tensor input)

Description copied from interface: Layer

Forward propagation (or forward pass) through the layer.

Parameters:

input - the input tensor.

Returns:

the output tensor.