Package smile.deep.layer

Class LinearLayer

java.lang.Object

smile.deep.layer.LinearLayer

All Implemented Interfaces:

Function<Tensor,Tensor>, Layer

public class LinearLayer
extends Object
implements Layer

A fully connected linear layer.

Constructor Summary

Constructors

Constructor	Description
LinearLayer(int in, int out)	Constructor.
LinearLayer(int in, int out, boolean bias)	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, to, to

Constructor Details

LinearLayer

public LinearLayer(int in,
 int out)

Constructor.

Parameters:

in - the number of input features.

out - the number of output features.

LinearLayer

public LinearLayer(int in,
 int out,
 boolean bias)

Constructor.

Parameters:

in - the number of input features.

out - the number of output features.

bias - If false, the layer will not learn an additive bias.

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.