Package smile.deep.layer

Class AdaptiveAvgPool2dLayer

java.lang.Object

smile.deep.layer.AdaptiveAvgPool2dLayer

All Implemented Interfaces:

Function<Tensor,Tensor>, Layer

public class AdaptiveAvgPool2dLayer
extends Object
implements Layer

An adaptive average pooling that reduces a tensor by combining cells. The output size should be specified and the stride and kernel-size are automatically selected to adapt to the needs.

Constructor Summary

Constructors

Constructor	Description
AdaptiveAvgPool2dLayer(int size)	Constructor.
AdaptiveAvgPool2dLayer(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 java.lang.Object

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

Methods inherited from interface java.util.function.Function

andThen, compose

Methods inherited from interface smile.deep.layer.Layer

apply, isTraining

Constructor Details

AdaptiveAvgPool2dLayer

public AdaptiveAvgPool2dLayer(int size)

Constructor.

Parameters:

size - the output size.

AdaptiveAvgPool2dLayer

public AdaptiveAvgPool2dLayer(int height,
 int width)

Constructor.

Parameters:

height - the output height.

width - the output 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.