Package | Description |
---|---|
org.nd4j.autodiff.samediff.ops | |
org.nd4j.linalg.api.ops.impl.layers.convolution | |
org.nd4j.linalg.factory.ops |
Modifier and Type | Method and Description |
---|---|
SDVariable |
SDCNN.avgPooling2d(SDVariable input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - average pooling 2d
|
SDVariable |
SDCNN.avgPooling2d(String name,
SDVariable input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - average pooling 2d
|
SDVariable |
SDCNN.maxPooling2d(SDVariable input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - max pooling 2d
|
SDVariable |
SDCNN.maxPooling2d(String name,
SDVariable input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - max pooling 2d
|
SDVariable[] |
SDCNN.maxPoolWithArgmax(SDVariable input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - Max pooling on the input and outputs both max values and indices
|
SDVariable[] |
SDCNN.maxPoolWithArgmax(String[] names,
SDVariable input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - Max pooling on the input and outputs both max values and indices
|
Modifier and Type | Field and Description |
---|---|
protected Pooling2DConfig |
Pooling2D.config |
protected Pooling2DConfig |
AvgPooling2D.config |
protected Pooling2DConfig |
MaxPoolWithArgmax.config |
protected Pooling2DConfig |
MaxPooling2D.config |
Modifier and Type | Method and Description |
---|---|
INDArray |
NDCNN.avgPooling2d(INDArray input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - average pooling 2d
|
INDArray |
NDCNN.maxPooling2d(INDArray input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - max pooling 2d
|
INDArray[] |
NDCNN.maxPoolWithArgmax(INDArray input,
Pooling2DConfig Pooling2DConfig)
2D Convolution layer operation - Max pooling on the input and outputs both max values and indices
|
Copyright © 2020. All rights reserved.