-
Constructor Summary
Constructors
-
Method Summary
org.bytedeco.pytorch.Module
Returns the PyTorch Module object.
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
-
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
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.