Dropout

class Dropout(dropoutPercent: Float) : Layer<Dropout> , LayerWithInferenceMode

Functions

cpu
Link copied to clipboard
open override fun cpu(): Layer<Dropout>
getSingleInput
Link copied to clipboard
open fun getSingleInput(inputs: Array<out DTensor>): DTensor

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

gpu
Link copied to clipboard
open override fun gpu(): Layer<Dropout>
invoke
Link copied to clipboard
open operator override fun invoke(vararg inputs: DTensor): DTensor
operator fun invoke(input: DTensor, random: Random): DTensor
to
Link copied to clipboard
open fun to(device: Device): OnDevice

Properties

dropoutPercent
Link copied to clipboard
val dropoutPercent: Float
inferenceMode
Link copied to clipboard
open override val inferenceMode: Layer<*>