Class SqueezeExcitation

java.lang.Object

smile.deep.layer.LayerBlock

smile.vision.layer.SqueezeExcitation

All Implemented Interfaces:

Function<Tensor,Tensor>, Layer

public class SqueezeExcitation
extends LayerBlock

Squeeze-and-Excitation block from "Squeeze-and-Excitation Networks".

Field Summary

Fields inherited from class LayerBlock

device, dtype, module

Constructor Summary

Constructors

Constructor	Description
SqueezeExcitation(int inputChannels, int squeezeChannels)	Constructor.
SqueezeExcitation(int inputChannels, int squeezeChannels, ActivationFunction delta, ActivationFunction sigma)	Constructor.

Method Summary

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

Methods inherited from class LayerBlock

add, add, asTorch, device, dtype, eval, isTraining, load, save, to, to, toString, train

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

Constructor Details

SqueezeExcitation

public SqueezeExcitation(int inputChannels,
 int squeezeChannels)

Constructor.

Parameters:

inputChannels - the number of channels in the input image.

squeezeChannels - the number of squeeze channels.

SqueezeExcitation

public SqueezeExcitation(int inputChannels,
 int squeezeChannels,
 ActivationFunction delta,
 ActivationFunction sigma)

Constructor.

Parameters:

inputChannels - the number of channels in the input image.

squeezeChannels - the number of squeeze channels.

delta - the delta activation function.

sigma - the sigma activation function.

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.