clj-djl.nn — clj-djl 0.1.7

add clj

(add net block)

batch-flatten clj

(batch-flatten array & more)

batch-flatten-block clj

(batch-flatten-block & more)

batchnorm-block clj

(batchnorm-block)

(batchnorm-block {:keys [axis center epsilon momentum scale]})

build clj

(build builder)

clear clj

(clear block)

cov2d-block clj

(cov2d-block {:keys [kernel-shape filters bias dilation groups padding stride]})

dropout clj

(dropout {:keys [rate]})

elu clj

(elu data alpha)

Applies ELU(Exponential Linear Unit) activation on the input NDArray or NDList

elu-block clj

(elu-block alpha)

Creates a LambdaBlock that applies the ELU activation function in its forward function

ELU <- (if (> x 0) x (* alpha (- (pow e x) 1)))

forward clj

(forward block inputs)

(forward block paramstore inputs labels-or-training? & [params])

gelu clj

(gelu data)

Applies GELU(Gausian Error Linear Unit) activation on the input NDArray or NDList

gelu-block clj

(gelu-block)

Creates a LambdaBlock that applies the GELU activation function in its forward function

get-parameters clj

(get-parameters block)

identity-block clj

(identity-block)

initialize clj

(initialize block manager datatype- & input-shapes)

leaky-relu clj

(leaky-relu data alpha)

leaky-relu-block clj

(leaky-relu-block alpha)

Create a LamdaBlock with LeakyReLU as forward function:

LeakyRelu = (if (>= x 0) x (* neg_slope x))

linear clj

(linear {:keys [bias units]})

linear-block clj

linear-builder clj

(linear-builder)

mish clj

(mish data)

Applies Mish activation on the input NDArray or NDList

mish-block clj

(mish-block)

Creates a LambdaBlock that applies the Mish activation function in its forward function

new-linear-builder clj

new-normal-initializer clj

normal-initializer clj

(normal-initializer)

(normal-initializer sigma)

opt-bias clj

(opt-bias builder bias)

prelu-block clj

(prelu-block)

Creates a LambdaBlock that applies the PreLU activation function in its forward function, the neg_slope is learnt during training

relu clj

(relu data)

relu-block clj

(relu-block)

selu clj

(selu data)

Applies SELU(Scaled Exponential Linear Unit) activation on the input NDArray or NDList

selu-block clj

(selu-block)

Creates a LambdaBlock that applies the SELU activation function in its forward function

SELU <- (* lambda (if (> x 0) x (* alpha (- (pow e x) 1)))), where lamda is 1.0507009873554804934193349852946 and alpha is 1.6732632423543772848170429916717

sequential clj

(sequential)

(sequential {:keys [blocks initializer parameter]})

sequential-block clj

set-initializer clj

(set-initializer net initializer parameter)

set-units clj

(set-units builder unit)

sigmoid clj

(sigmoid data)

sigmoid-block clj

(sigmoid-block)

softplus clj

(softplus data)

softplus-block clj

(softplus-block)

swish clj

(swish data beta)

Applies Swish activation on the input NDArray or NDList

swish-block clj

(swish-block beta)

Creates a LambdaBlock that applies the Swish activation function in its forward function

tanh clj

(tanh data)

tanh-block clj

(tanh-block)