Linear

constructor(inFeatures: Int, outFeatures: Int, name: String = "Linear", initWeights: Tensor<T, V>, initBias: Tensor<T, V>, trainable: Boolean = true)(source)

Parameters

inFeatures

Number of input features

outFeatures

Number of output features

name

Name of the module

initWeights

Initial weights

initBias

Initial bias