org.deeplearning4j.nn.layers

Class Layer.Builder

java.lang.Object

org.deeplearning4j.nn.layers.Layer.Builder

Enclosing class:

Layer

public static class Layer.Builder
extends Object

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.api.ndarray.INDArray	b
protected NeuralNetConfiguration	conf
protected org.nd4j.linalg.api.ndarray.INDArray	input
protected org.nd4j.linalg.api.ndarray.INDArray	W

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
Layer	build()
Layer.Builder	conf(NeuralNetConfiguration conf)
Layer.Builder	withBias(org.nd4j.linalg.api.ndarray.INDArray b)
Layer.Builder	withInput(org.nd4j.linalg.api.ndarray.INDArray input)
Layer.Builder	withWeights(org.nd4j.linalg.api.ndarray.INDArray W)

Methods inherited from class java.lang.Object

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

Field Detail

W

protected org.nd4j.linalg.api.ndarray.INDArray W

b

protected org.nd4j.linalg.api.ndarray.INDArray b

input

protected org.nd4j.linalg.api.ndarray.INDArray input

conf

protected NeuralNetConfiguration conf

Constructor Detail

Builder

public Builder()

Method Detail

conf

public Layer.Builder conf(NeuralNetConfiguration conf)

withWeights

public Layer.Builder withWeights(org.nd4j.linalg.api.ndarray.INDArray W)

withBias

public Layer.Builder withBias(org.nd4j.linalg.api.ndarray.INDArray b)

withInput

public Layer.Builder withInput(org.nd4j.linalg.api.ndarray.INDArray input)

build

public Layer build()