ReluLayer

Functions

cpu

open override fun cpu(): Layer<ReluLayer>

getSingleInput

Helper to check that the layer was called with a single input. Returns that input if successful, else errors.

gpu

open override fun gpu(): Layer<ReluLayer>

invoke

open operator override fun invoke(vararg inputs: DTensor): DTensor

open operator override fun invoke(input: DTensor): DTensor

open fun to(device: Device): OnDevice