public class Convolution extends Object
Modifier and Type | Class and Description |
---|---|
static class |
Convolution.Type |
Modifier and Type | Method and Description |
---|---|
static INDArray |
col2im(INDArray col,
INDArray z,
int sy,
int sx,
int ph,
int pw,
int h,
int w) |
static INDArray |
col2im(INDArray col,
int[] stride,
int[] padding,
int height,
int width) |
static INDArray |
col2im(INDArray col,
int sy,
int sx,
int ph,
int pw,
int h,
int w)
Rearrange matrix
columns into blocks
|
static INDArray |
conv2d(IComplexNDArray input,
IComplexNDArray kernel,
Convolution.Type type) |
static INDArray |
conv2d(INDArray input,
INDArray kernel,
Convolution.Type type)
2d convolution (aka the last 2 dimensions
|
static IComplexNDArray |
convn(IComplexNDArray input,
IComplexNDArray kernel,
Convolution.Type type)
ND Convolution
|
static IComplexNDArray |
convn(IComplexNDArray input,
IComplexNDArray kernel,
Convolution.Type type,
int[] axes)
ND Convolution
|
static INDArray |
convn(INDArray input,
INDArray kernel,
Convolution.Type type)
ND Convolution
|
static INDArray |
convn(INDArray input,
INDArray kernel,
Convolution.Type type,
int[] axes)
ND Convolution
|
static INDArray |
im2col(INDArray img,
int[] kernel,
int[] stride,
int[] padding) |
static INDArray |
im2col(INDArray img,
int kh,
int kw,
int sy,
int sx,
int ph,
int pw,
boolean isSameMode)
Implement column formatted images
|
static INDArray |
im2col(INDArray img,
int kh,
int kw,
int sy,
int sx,
int ph,
int pw,
boolean isSameMode,
INDArray out) |
static INDArray |
im2col(INDArray img,
int kh,
int kw,
int sy,
int sx,
int ph,
int pw,
int pval,
boolean isSameMode)
Implement column formatted images
|
static int |
outSize(int size,
int k,
int s,
int p,
boolean coverAll)
The out size for a convolution
|
static INDArray |
pooling2D(INDArray img,
int kh,
int kw,
int sy,
int sx,
int ph,
int pw,
boolean isSameMode,
Pooling2D.Pooling2DType type,
double extra,
int virtualHeight,
int virtualWidth,
INDArray out) |
public static INDArray col2im(INDArray col, int[] stride, int[] padding, int height, int width)
col
- stride
- padding
- height
- width
- public static INDArray col2im(INDArray col, int sy, int sx, int ph, int pw, int h, int w)
col
- the column
transposed image to convertsy
- stride ysx
- stride xph
- padding heightpw
- padding widthh
- heightw
- widthpublic static INDArray col2im(INDArray col, INDArray z, int sy, int sx, int ph, int pw, int h, int w)
public static INDArray im2col(INDArray img, int[] kernel, int[] stride, int[] padding)
img
- kernel
- stride
- padding
- public static INDArray im2col(INDArray img, int kh, int kw, int sy, int sx, int ph, int pw, boolean isSameMode)
img
- the image to processkh
- the kernel heightkw
- the kernel widthsy
- the stride along ysx
- the stride along xph
- the padding widthpw
- the padding heightisSameMode
- whether to cover the whole image or notpublic static INDArray im2col(INDArray img, int kh, int kw, int sy, int sx, int ph, int pw, boolean isSameMode, INDArray out)
public static INDArray pooling2D(INDArray img, int kh, int kw, int sy, int sx, int ph, int pw, boolean isSameMode, Pooling2D.Pooling2DType type, double extra, int virtualHeight, int virtualWidth, INDArray out)
public static INDArray im2col(INDArray img, int kh, int kw, int sy, int sx, int ph, int pw, int pval, boolean isSameMode)
img
- the image to processkh
- the kernel heightkw
- the kernel widthsy
- the stride along ysx
- the stride along xph
- the padding widthpw
- the padding heightpval
- the padding value (not used)isSameMode
- whether padding mode is 'same'public static int outSize(int size, int k, int s, int p, boolean coverAll)
size
- k
- s
- p
- coverAll
- public static INDArray conv2d(INDArray input, INDArray kernel, Convolution.Type type)
input
- the input to opkernel
- the kernel to convolve withtype
- public static INDArray conv2d(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type)
input
- kernel
- type
- public static INDArray convn(INDArray input, INDArray kernel, Convolution.Type type, int[] axes)
input
- the input to opkernel
- the kerrnel to op withtype
- the type of convolutionaxes
- the axes to do the convolution alongpublic static IComplexNDArray convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type, int[] axes)
input
- the input to opkernel
- the kernel to op withtype
- the type of convolutionaxes
- the axes to do the convolution alongpublic static INDArray convn(INDArray input, INDArray kernel, Convolution.Type type)
input
- the input to opkernel
- the kernel to op withtype
- the type of convolutionpublic static IComplexNDArray convn(IComplexNDArray input, IComplexNDArray kernel, Convolution.Type type)
input
- the input to opkernel
- the kernel to op withtype
- the type of convolutionCopyright © 2017. All rights reserved.