org.nd4j.linalg.api.ops.impl.layers.convolution.config

Class Conv2DConfig

java.lang.Object

org.nd4j.linalg.api.ops.impl.layers.convolution.config.BaseConvolutionConfig

org.nd4j.linalg.api.ops.impl.layers.convolution.config.Conv2DConfig

public class Conv2DConfig
extends BaseConvolutionConfig

Field Summary

Fields

Modifier and Type	Field and Description
static int	CAUSAL
static String	NCHW
static String	NHWC
static int	SAME
static int	VALID

Constructor Summary

Constructors

Constructor and Description
Conv2DConfig(long kH, long kW, long sH, long sW, long pH, long pW, long dH, long dW, PaddingMode paddingMode, String dataFormat, WeightsFormat weightsFormat)

Method Summary

Modifier and Type	Method and Description
boolean	isNHWC()
void	isNHWC(boolean isNHWC)
Map<String,Object>	toProperties()
protected void	validate()

Methods inherited from class org.nd4j.linalg.api.ops.impl.layers.convolution.config.BaseConvolutionConfig

getValue, setValueFor

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

NCHW

public static final String NCHW

See Also:

Constant Field Values

NHWC

public static final String NHWC

See Also:

Constant Field Values

VALID

public static final int VALID

See Also:

Constant Field Values

SAME

public static final int SAME

See Also:

Constant Field Values

CAUSAL

public static final int CAUSAL

See Also:

Constant Field Values

Constructor Detail

Conv2DConfig

public Conv2DConfig(long kH,
                    long kW,
                    long sH,
                    long sW,
                    long pH,
                    long pW,
                    long dH,
                    long dW,
                    PaddingMode paddingMode,
                    String dataFormat,
                    WeightsFormat weightsFormat)

Method Detail

isNHWC

public boolean isNHWC()

isNHWC

public void isNHWC(boolean isNHWC)

toProperties

public Map<String,Object> toProperties()

Specified by:

toProperties in class BaseConvolutionConfig

validate

protected void validate()

Specified by:

validate in class BaseConvolutionConfig