public final class Pool
extends java.lang.Object
Block
and methods for different pooling functions.Modifier and Type | Method and Description |
---|---|
static NDArray |
avgPool1d(NDArray input,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode,
boolean countIncludePad)
Performs 1-D Avg Pooling on the input.
|
static Block |
avgPool1dBlock(Shape kernelShape)
Creates a
LambdaBlock that applies the avgPool1dBlock pooling function in its forward function. |
static Block |
avgPool1dBlock(Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the avgPool1dBlock pooling function in its forward function. |
static Block |
avgPool1dBlock(Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the avgPool1dBlock pooling function in its forward function. |
static Block |
avgPool1dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the avgPool1dBlock pooling function in its forward function. |
static Block |
avgPool1dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode,
boolean countIncludePad)
Creates a
LambdaBlock that applies the avgPool1dBlock pooling function in its forward function. |
static NDArray |
avgPool2d(NDArray input,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode,
boolean countIncludePad)
Performs 2-D Avg Pooling on the input.
|
static Block |
avgPool2dBlock(Shape kernelShape)
Creates a
LambdaBlock that applies the avgPool2dBlock pooling function in its forward function. |
static Block |
avgPool2dBlock(Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the avgPool2dBlock pooling function in its forward function. |
static Block |
avgPool2dBlock(Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the avgPool2dBlock pooling function in its forward function. |
static Block |
avgPool2dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the avgPool2dBlock pooling function in its forward function. |
static Block |
avgPool2dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode,
boolean countIncludePad)
Creates a
LambdaBlock that applies the avgPool2dBlock pooling function in its forward function. |
static NDArray |
avgPool3d(NDArray input,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode,
boolean countIncludePad)
Performs 3-D Avg Pooling on the input.
|
static Block |
avgPool3dBlock(Shape kernelShape)
Creates a
LambdaBlock that applies the avgPool3dBlock pooling function in its forward function. |
static Block |
avgPool3dBlock(Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the avgPool3dBlock pooling function in its forward function. |
static Block |
avgPool3dBlock(Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the avgPool3dBlock pooling function in its forward function. |
static Block |
avgPool3dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the avgPool3dBlock pooling function in its forward function. |
static Block |
avgPool3dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode,
boolean countIncludePad)
Creates a
LambdaBlock that applies the avgPool3dBlock pooling function in its forward function. |
static NDArray |
globalAvgPool1d(NDArray input)
Performs 1-D Global Avg Pooling on the input.
|
static Block |
globalAvgPool1dBlock()
Creates a
LambdaBlock that applies the globalAvgPool1d pooling function. |
static NDArray |
globalAvgPool2d(NDArray input)
Performs 2-D Global Avg Pooling on the input.
|
static Block |
globalAvgPool2dBlock()
Creates a
LambdaBlock that applies the globalAvgPool2d pooling function. |
static NDArray |
globalAvgPool3d(NDArray input)
Performs 3-D Global Avg Pooling on the input.
|
static Block |
globalAvgPool3dBlock()
Creates a
LambdaBlock that applies the globalAvgPool3d pooling function. |
static NDArray |
globalLpPool1d(NDArray input,
float normType)
Performs 1-D Global LP Pooling on the input.
|
static Block |
globalLpPool1dBlock(float normType)
Creates a
LambdaBlock that applies the globalLpPool1d pooling function. |
static NDArray |
globalLpPool2d(NDArray input,
float normType)
Performs 2-D Global LP Pooling on the input.
|
static Block |
globalLpPool2dBlock(float normType)
Creates a
LambdaBlock that applies the globalLpPool2d pooling function. |
static NDArray |
globalLpPool3d(NDArray input,
float normType)
Performs 3-D Global LP Pooling on the input.
|
static Block |
globalLpPool3dBlock(float normType)
Creates a
LambdaBlock that applies the globalLpPool3d pooling function. |
static NDArray |
globalMaxPool1d(NDArray input)
Performs 1-D Global Max Pooling on the input.
|
static Block |
globalMaxPool1dBlock()
Creates a
LambdaBlock that applies the globalmaxPool1dBlock pooling function. |
static NDArray |
globalMaxPool2d(NDArray input)
Performs 2-D Global Max Pooling on the input.
|
static Block |
globalMaxPool2dBlock()
Creates a
LambdaBlock that applies the globalmaxPool2dBlock pooling function. |
static NDArray |
globalMaxPool3d(NDArray input)
Performs 3-D Global Max Pooling on the input.
|
static Block |
globalMaxPool3dBlock()
Creates a
LambdaBlock that applies the globalmaxPool3dBlock pooling function. |
static NDArray |
lpPool1d(NDArray input,
float normType,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Performs 1-D LP Pooling on the input.
|
static Block |
lpPool1dBlock(float normType,
Shape kernelShape)
Creates a
LambdaBlock that applies the lpPool1dBlock pooling function in its forward function. |
static Block |
lpPool1dBlock(float normType,
Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the lpPool1dBlock pooling function in its forward function. |
static Block |
lpPool1dBlock(float normType,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the lpPool1dBlock pooling function in its forward function. |
static NDArray |
lpPool2d(NDArray input,
float normType,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Performs 2-D LP Pooling on the input.
|
static Block |
lpPool2dBlock(float normType,
Shape kernelShape)
Creates a
LambdaBlock that applies the lpPool2dBlock pooling function in its forward function. |
static Block |
lpPool2dBlock(float normType,
Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the lpPool2dBlock pooling function in its forward function. |
static Block |
lpPool2dBlock(float normType,
Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the lpPool2dBlock pooling function in its forward function. |
static Block |
lpPool2dBlock(float normType,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the lpPool2dBlock pooling function in its forward function. |
static NDArray |
lpPool3d(NDArray input,
float normType,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Performs 3-D LP Pooling on the input.
|
static Block |
lpPool3dBlock(float normType,
Shape kernelShape)
Creates a
LambdaBlock that applies the lpPool3dBlock pooling function in its forward function. |
static Block |
lpPool3dBlock(float normType,
Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the LpPoo3D pooling function in its forward function. |
static Block |
lpPool3dBlock(float normType,
Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the lpPool3dBlock pooling function in its forward function. |
static Block |
lpPool3dBlock(float normType,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the lpPool3dBlock pooling function in its forward function. |
static NDArray |
maxPool1d(NDArray input,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Performs 1-D Max Pooling on the input.
|
static Block |
maxPool1dBlock(Shape kernelShape)
Creates a
LambdaBlock that applies the maxPool1dBlock pooling function in its forward function. |
static Block |
maxPool1dBlock(Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the maxPool1dBlock pooling function in its forward function. |
static Block |
maxPool1dBlock(Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the maxPool1dBlock pooling function in its forward function. |
static Block |
maxPool1dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the maxPool1d pooling function in its forward function. |
static NDArray |
maxPool2d(NDArray input,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Performs 2-D Max Pooling on the input.
|
static Block |
maxPool2dBlock(Shape kernelShape)
Creates a
LambdaBlock that applies the maxPool2dBlock pooling function in its forward function. |
static Block |
maxPool2dBlock(Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the maxPool2dBlock pooling function in its forward function. |
static Block |
maxPool2dBlock(Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the maxPool2dBlock pooling function in its forward function. |
static Block |
maxPool2dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the maxPool2dBlock pooling function in its forward function. |
static NDArray |
maxPool3d(NDArray input,
Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Performs 3-D Max Pooling on the input.
|
static Block |
maxPool3dBlock(Shape kernelShape)
Creates a
LambdaBlock that applies the maxPool3dBlock pooling function in its forward function. |
static Block |
maxPool3dBlock(Shape kernelShape,
Shape stride)
Creates a
LambdaBlock that applies the maxPool3dBlock pooling function in its forward function. |
static Block |
maxPool3dBlock(Shape kernelShape,
Shape stride,
Shape padding)
Creates a
LambdaBlock that applies the maxPool3dBlock pooling function in its forward function. |
static Block |
maxPool3dBlock(Shape kernelShape,
Shape stride,
Shape padding,
boolean ceilMode)
Creates a
LambdaBlock that applies the maxPool3dBlock pooling function in its forward function. |
public static NDArray maxPool1d(NDArray input, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
input
- the NDArray on which max pooling is performedkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.public static NDArray maxPool2d(NDArray input, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
input
- the NDArray on which max pooling is performedkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.public static NDArray maxPool3d(NDArray input, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
input
- the NDArray on which max pooling is performedkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.public static NDArray globalMaxPool1d(NDArray input)
input
- the NDArray on which max pooling is performedpublic static NDArray globalMaxPool2d(NDArray input)
input
- the NDArray on which max pooling is performedpublic static NDArray globalMaxPool3d(NDArray input)
input
- the NDArray on which max pooling is performedpublic static NDArray avgPool1d(NDArray input, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode, boolean countIncludePad)
input
- the NDArray on which average pooling is performedkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.countIncludePad
- whether to include padding for calculationspublic static NDArray avgPool2d(NDArray input, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode, boolean countIncludePad)
input
- the NDArray on which average pooling is performedkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.countIncludePad
- whether to include padding for calculationspublic static NDArray avgPool3d(NDArray input, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode, boolean countIncludePad)
input
- the NDArray on which average pooling is performedkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.countIncludePad
- whether to include padding for calculationspublic static NDArray globalAvgPool1d(NDArray input)
input
- the NDArray on which average pooling is performedpublic static NDArray globalAvgPool2d(NDArray input)
input
- the NDArray on which average pooling is performedpublic static NDArray globalAvgPool3d(NDArray input)
input
- the NDArray on which average pooling is performedpublic static NDArray lpPool1d(NDArray input, float normType, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
input
- the NDArray on which LP pooling is performednormType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.public static NDArray lpPool2d(NDArray input, float normType, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
input
- the NDArray on which LP pooling is performednormType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.public static NDArray lpPool3d(NDArray input, float normType, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
input
- the NDArray on which LP pooling is performednormType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- the stride to be used for each dimensionpadding
- the padding to be set in each dimensionceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.public static NDArray globalLpPool1d(NDArray input, float normType)
input
- the NDArray on which LP pooling is performednormType
- float value indicating normpublic static NDArray globalLpPool2d(NDArray input, float normType)
input
- the NDArray on which LP pooling is performednormType
- float value indicating normpublic static NDArray globalLpPool3d(NDArray input, float normType)
input
- the NDArray on which LP pooling is performednormType
- float value indicating normpublic static Block maxPool1dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the maxPool1d
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the maxPool1dBlock
activation functionpublic static Block maxPool1dBlock(Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the maxPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the maxPool1dBlock
activation functionpublic static Block maxPool1dBlock(Shape kernelShape, Shape stride)
LambdaBlock
that applies the maxPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the maxPool1dBlock
activation functionpublic static Block maxPool1dBlock(Shape kernelShape)
LambdaBlock
that applies the maxPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the maxPool1dBlock
activation functionpublic static Block maxPool2dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the maxPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the maxPool2dBlock
activation functionpublic static Block maxPool2dBlock(Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the maxPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the maxPool2dBlock
activation functionpublic static Block maxPool2dBlock(Shape kernelShape, Shape stride)
LambdaBlock
that applies the maxPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the maxPool2dBlock
activation functionpublic static Block maxPool2dBlock(Shape kernelShape)
LambdaBlock
that applies the maxPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the maxPool2dBlock
activation functionpublic static Block maxPool3dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the maxPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the maxPool3dBlock
activation functionpublic static Block maxPool3dBlock(Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the maxPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the maxPool3dBlock
activation functionpublic static Block maxPool3dBlock(Shape kernelShape, Shape stride)
LambdaBlock
that applies the maxPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the maxPool3dBlock
activation functionpublic static Block maxPool3dBlock(Shape kernelShape)
LambdaBlock
that applies the maxPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the maxPool3dBlock
activation functionpublic static Block globalMaxPool1dBlock()
LambdaBlock
that applies the globalmaxPool1dBlock
pooling function.LambdaBlock
that applies the globalmaxPool1dBlock
pooling functionpublic static Block globalMaxPool2dBlock()
LambdaBlock
that applies the globalmaxPool2dBlock
pooling function.LambdaBlock
that applies the globalmaxPool2dBlock
pooling functionpublic static Block globalMaxPool3dBlock()
LambdaBlock
that applies the globalmaxPool3dBlock
pooling function.LambdaBlock
that applies the globalmaxPool3dBlock
pooling functionpublic static Block avgPool1dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode, boolean countIncludePad)
LambdaBlock
that applies the avgPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.countIncludePad
- Boolean indicating whether to include padding for calculationsLambdaBlock
that applies the avgPool1dBlock
activation functionpublic static Block avgPool1dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the avgPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the avgPool1dBlock
activation functionpublic static Block avgPool1dBlock(Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the avgPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the avgPool1dBlock
activation functionpublic static Block avgPool1dBlock(Shape kernelShape, Shape stride)
LambdaBlock
that applies the avgPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the avgPool1dBlock
activation functionpublic static Block avgPool1dBlock(Shape kernelShape)
LambdaBlock
that applies the avgPool1dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the avgPool1dBlock
activation functionpublic static Block avgPool2dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode, boolean countIncludePad)
LambdaBlock
that applies the avgPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.countIncludePad
- Boolean indicating whether to include padding for calculationsLambdaBlock
that applies the avgPool2dBlock
activation functionpublic static Block avgPool2dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the avgPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the avgPool2dBlock
activation functionpublic static Block avgPool2dBlock(Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the avgPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the avgPool2dBlock
activation functionpublic static Block avgPool2dBlock(Shape kernelShape, Shape stride)
LambdaBlock
that applies the avgPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the avgPool2dBlock
activation functionpublic static Block avgPool2dBlock(Shape kernelShape)
LambdaBlock
that applies the avgPool2dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the avgPool2dBlock
activation functionpublic static Block avgPool3dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode, boolean countIncludePad)
LambdaBlock
that applies the avgPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.countIncludePad
- Boolean indicating whether to include padding for calculationsLambdaBlock
that applies the avgPool3dBlock
activation functionpublic static Block avgPool3dBlock(Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the avgPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the avgPool3dBlock
activation functionpublic static Block avgPool3dBlock(Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the avgPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the avgPool3dBlock
activation functionpublic static Block avgPool3dBlock(Shape kernelShape, Shape stride)
LambdaBlock
that applies the avgPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the avgPool3dBlock
activation functionpublic static Block avgPool3dBlock(Shape kernelShape)
LambdaBlock
that applies the avgPool3dBlock
pooling function in its forward function.kernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the avgPool3dBlock
activation functionpublic static Block globalAvgPool1dBlock()
LambdaBlock
that applies the globalAvgPool1d
pooling function.LambdaBlock
that applies the globalAvgPool1d
pooling functionpublic static Block globalAvgPool2dBlock()
LambdaBlock
that applies the globalAvgPool2d
pooling function.LambdaBlock
that applies the globalAvgPool2d
pooling functionpublic static Block globalAvgPool3dBlock()
LambdaBlock
that applies the globalAvgPool3d
pooling function.LambdaBlock
that applies the globalAvgPool3d
pooling functionpublic static Block lpPool1dBlock(float normType, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the lpPool1dBlock
pooling function in its forward function.normType
- integer indicating pValuekernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- padding of pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the lpPool1dBlock
activation functionpublic static Block lpPool1dBlock(float normType, Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the lpPool1dBlock
pooling function in its forward function.normType
- integer indicating pValuekernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- padding of pooling layerLambdaBlock
that applies the lpPool1dBlock
activation functionpublic static Block lpPool1dBlock(float normType, Shape kernelShape)
LambdaBlock
that applies the lpPool1dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the lpPool1dBlock
activation functionpublic static Block lpPool2dBlock(float normType, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the lpPool2dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the lpPool2dBlock
activation functionpublic static Block lpPool2dBlock(float normType, Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the lpPool2dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the lpPool2dBlock
activation functionpublic static Block lpPool2dBlock(float normType, Shape kernelShape, Shape stride)
LambdaBlock
that applies the lpPool2dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the lpPool2dBlock
activation functionpublic static Block lpPool2dBlock(float normType, Shape kernelShape)
LambdaBlock
that applies the lpPool2dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the lpPool2dBlock
activation functionpublic static Block lpPool3dBlock(float normType, Shape kernelShape, Shape stride, Shape padding, boolean ceilMode)
LambdaBlock
that applies the lpPool3dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerceilMode
- when true, will use ceil instead of floor in the formula to compute the
output shape. The formula is f(x, k, p, s) = floor((x+2*p-k)/s)+1
.LambdaBlock
that applies the lpPool3dBlock
activation functionpublic static Block lpPool3dBlock(float normType, Shape kernelShape, Shape stride, Shape padding)
LambdaBlock
that applies the lpPool3dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerpadding
- pad of the pooling layerLambdaBlock
that applies the lpPool3dBlock
activation functionpublic static Block lpPool3dBlock(float normType, Shape kernelShape, Shape stride)
LambdaBlock
that applies the LpPoo3D
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedstride
- stride of pooling layerLambdaBlock
that applies the lpPool3dBlock
activation functionpublic static Block lpPool3dBlock(float normType, Shape kernelShape)
LambdaBlock
that applies the lpPool3dBlock
pooling function in its forward function.normType
- float value indicating normkernelShape
- the shape of the kernel to be usedLambdaBlock
that applies the lpPool3dBlock
activation functionpublic static Block globalLpPool1dBlock(float normType)
LambdaBlock
that applies the globalLpPool1d
pooling function.normType
- float value indicating normLambdaBlock
that applies the globalLpPool1d
pooling functionpublic static Block globalLpPool2dBlock(float normType)
LambdaBlock
that applies the globalLpPool2d
pooling function.normType
- float value indicating normLambdaBlock
that applies the globalLpPool2d
pooling functionpublic static Block globalLpPool3dBlock(float normType)
LambdaBlock
that applies the globalLpPool3d
pooling function.normType
- float value indicating normLambdaBlock
that applies the globalLpPool3d
pooling function