org.deeplearning4j.nn.conf.layers

Class Upsampling2D.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.BaseUpsamplingLayer.UpsamplingBuilder<Upsampling2D.Builder>

org.deeplearning4j.nn.conf.layers.Upsampling2D.Builder

Enclosing class:

Upsampling2D

public static class Upsampling2D.Builder
extends BaseUpsamplingLayer.UpsamplingBuilder<Upsampling2D.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected CNN2DFormat	format

Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseUpsamplingLayer.UpsamplingBuilder

size

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder(int size)

Method Summary

Modifier and Type	Method and Description
Upsampling2D	build()
Upsampling2D.Builder	dataFormat(CNN2DFormat format) Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).
void	setSize(int... size)
Upsampling2D.Builder	size(int size) Upsampling size int, used for both height and width
Upsampling2D.Builder	size(int[] size) Upsampling size array

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

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

Field Detail

format

protected CNN2DFormat format

Constructor Detail

Builder

public Builder(int size)

Method Detail

dataFormat

public Upsampling2D.Builder dataFormat(CNN2DFormat format)

Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last). See CNN2DFormat for more details.
Default: NCHW

Parameters:

format - Format for activations (in and out)

size

public Upsampling2D.Builder size(int size)

Upsampling size int, used for both height and width

Parameters:

size - upsampling size in height and width dimensions

size

public Upsampling2D.Builder size(int[] size)

Upsampling size array

Parameters:

size - upsampling size in height and width dimensions

build

public Upsampling2D build()

Specified by:

build in class Layer.Builder<Upsampling2D.Builder>

setSize

public void setSize(int... size)