| Package | Description |
|---|---|
| org.nd4j.autodiff.functions | |
| org.nd4j.autodiff.samediff | |
| org.nd4j.linalg.api.ops.impl.layers.convolution |
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
DifferentialFunctionFactory.avgPooling2d(SDVariable[] inputs,
Pooling2DConfig pooling2DConfig)
Average pooling 2d operation.
|
SDVariable |
DifferentialFunctionFactory.maxPooling2d(SDVariable[] inputs,
Pooling2DConfig pooling2DConfig)
Max pooling 2d operation.
|
| Modifier and Type | Method and Description |
|---|---|
SDVariable |
SameDiff.avgPooling2d(SDVariable[] inputs,
Pooling2DConfig pooling2DConfig)
Average pooling 2d operation.
|
SDVariable |
SameDiff.avgPooling2d(String name,
SDVariable[] inputs,
Pooling2DConfig pooling2DConfig)
Average pooling 2d operation.
|
SDVariable |
SameDiff.maxPooling2d(SDVariable[] inputs,
Pooling2DConfig pooling2DConfig)
Max pooling 2d operation.
|
SDVariable |
SameDiff.maxPooling2d(String name,
SDVariable[] inputs,
Pooling2DConfig pooling2DConfig)
Max pooling 2d operation.
|
| Modifier and Type | Field and Description |
|---|---|
protected Pooling2DConfig |
Pooling2D.config |
protected Pooling2DConfig |
MaxPooling2D.config |
protected Pooling2DConfig |
AvgPooling2D.config |
| Constructor and Description |
|---|
AvgPooling2D(SameDiff sameDiff,
SDVariable[] inputs,
INDArray[] arrayInputs,
INDArray[] arrayOutputs,
Pooling2DConfig config) |
MaxPooling2D(SameDiff sameDiff,
SDVariable[] inputs,
INDArray[] arrayInputs,
INDArray[] arrayOutputs,
Pooling2DConfig config) |
Pooling2D(SameDiff sameDiff,
SDVariable[] inputs,
INDArray[] arrayInputs,
INDArray[] arrayOutputs,
Pooling2DConfig config) |
Pooling2DDerivative(SameDiff sameDiff,
SDVariable[] inputs,
INDArray[] arrayInputs,
INDArray[] arrayOutputs,
Pooling2DConfig config) |
Copyright © 2018. All rights reserved.