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,
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,
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)
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 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 valuecoverAll
- 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 © 2015. All Rights Reserved.