FullyConnectedLayer

java.lang.Object

smile.deep.layer.FullyConnectedLayer

All Implemented Interfaces:: Function<Tensor,Tensor>, Layer

public class FullyConnectedLayer extends Object implements Layer

A fully connected layer with nonlinear activation function.

Constructor Summary

Constructors

Constructor

Description

FullyConnectedLayer(int in, int out)

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
- FullyConnectedLayer
  
  public FullyConnectedLayer(int in, int out)
  
  Constructor.
  
  Parameters:
  
  in - the number of input features.
  
  out - the number of output features.
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.