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 coverAll)
Implement column formatted images
|
static INDArray |
im2col(INDArray img,
int kh,
int kw,
int sy,
int sx,
int ph,
int pw,
boolean coverAll,
INDArray out) |
static INDArray |
im2col(INDArray img,
int kh,
int kw,
int sy,
int sx,
int ph,
int pw,
int pval,
boolean coverAll)
Implement column formatted images
|
static int |
outSize(int size,
int k,
int s,
int p,
boolean coverAll)
The out size for a convolution
|
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 coverAll)
img
- the image to processkh
- the kernel heightkw
- the kernel widthsy
- the stride along ysx
- the stride along xph
- the padding widthpw
- the padding heightcoverAll
- 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 coverAll, INDArray out)
public static INDArray im2col(INDArray img, int kh, int kw, int sy, int sx, int ph, int pw, int pval, boolean coverAll)
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)coverAll
- whether to cover the whole image or notpublic 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 © 2016. All Rights Reserved.