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
-
-
Constructor Summary
Constructors Constructor 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
All Methods Instance Methods Concrete Methods Modifier and Type Method Description booleanisNHWC()voidisNHWC(boolean isNHWC)Map<String,Object>toProperties()protected voidvalidate()-
Methods inherited from class org.nd4j.linalg.api.ops.impl.layers.convolution.config.BaseConvolutionConfig
getValue, setValueFor
-
-
-
-
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:
toPropertiesin classBaseConvolutionConfig
-
validate
protected void validate()
- Specified by:
validatein classBaseConvolutionConfig
-
-