| Package | Description |
|---|---|
| org.deeplearning4j.nn.conf.layers |
| Modifier and Type | Method and Description |
|---|---|
PReLULayer.Builder |
PReLULayer.Builder.inputShape(long... shape)
Explicitly set input shape of incoming activations so that parameters can be initialized properly.
|
PReLULayer.Builder |
PReLULayer.Builder.sharedAxes(long... axes)
Set the broadcasting axes of PReLU's alpha parameter.
|
Copyright © 2021. All rights reserved.