Class AvgPool2dLayer

java.lang.Object

smile.deep.layer.ModuleLayer

smile.deep.layer.AvgPool2dLayer

All Implemented Interfaces:

Function<Tensor,Tensor>, Layer

public class AvgPool2dLayer
extends ModuleLayer

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

Constructor Summary

Constructors

Constructor	Description
AvgPool2dLayer(int kernel)	Constructor.
AvgPool2dLayer(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

AvgPool2dLayer

public AvgPool2dLayer(int kernel)

Constructor.

Parameters:

kernel - the window/kernel size.

AvgPool2dLayer

public AvgPool2dLayer(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.