- All Implemented Interfaces:
Function<Tensor,Tensor>
, Layer
public class MaxPool2dLayer
extends Object
implements Layer
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
-
Method Summary 
org.bytedeco.pytorch.Module
Returns the PyTorch Module object.
Forward propagation (or forward pass) through the layer.
Methods inherited from class java.lang.Object 
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
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 
-
asTorch 
public org.bytedeco.pytorch.Module asTorch()
Description copied from interface: Layer
Returns the PyTorch Module object.
- Specified by:
asTorch
in interface Layer
- Returns:
- the PyTorch Module object.
-
forward 
Description copied from interface: Layer
Forward propagation (or forward pass) through the layer.
- Specified by:
forward
in interface Layer
- Parameters:
input
- the input tensor.
- Returns:
- the output tensor.