MaxPool2dLayer

java.lang.Object

smile.deep.layer.MaxPool2dLayer

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

Constructor

Description

MaxPool2dLayer(int kernel)

Constructor.

MaxPool2dLayer(int height, int width)

Constructor.
Method Summary

Modifier and Type

Method

Description

org.bytedeco.pytorch.Module

asTorch()

Returns the PyTorch Module object.

Tensor

forward(Tensor input)

Forward propagation (or forward pass) through the layer.

Methods inherited from class Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface Function
andThen, compose

Methods inherited from interface Layer
apply, 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
- 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
  
  public Tensor forward(Tensor input)
  
  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.