Flatten

Flattens the input. Does not affect batch size.

Shape transform: (N, *) -> (N, )

Functions

cpu

open override fun cpu(): Layer<Flatten>

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<Flatten>

invoke

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

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

open fun to(device: Device): OnDevice