Package org.jetbrains.kotlinx.dl.api.core.layer.merge

Types

AbstractMerge
Link copied to clipboard
abstract class AbstractMerge(layerTypeName: String, name: String) : Layer
Content copied to clipboard

Layer that adds a list of inputs.

Add
Link copied to clipboard
class Add(name: String) : AbstractMerge
Content copied to clipboard

Layer that adds a list of inputs.

Average
Link copied to clipboard
class Average(name: String) : AbstractMerge
Content copied to clipboard

Layer that averages a list of inputs element-wise.

Concatenate
Link copied to clipboard
class Concatenate(axis: Int, name: String) : AbstractMerge, NoGradients
Content copied to clipboard

Layer that concatenates a list of inputs.

Maximum
Link copied to clipboard
class Maximum(name: String) : AbstractMerge
Content copied to clipboard

Layer that computes the maximum (element-wise) a list of inputs.

Minimum
Link copied to clipboard
class Minimum(name: String) : AbstractMerge
Content copied to clipboard

Layer that computes the minimum (element-wise) a list of inputs.

Multiply
Link copied to clipboard
class Multiply(name: String) : AbstractMerge
Content copied to clipboard

Layer that multiplies (element-wise) a list of inputs.

Subtract
Link copied to clipboard
class Subtract(name: String) : AbstractMerge
Content copied to clipboard

Layer that subtracts inputs, following from the previous.