Flatten

class Flatten<T : DType, V>(startDim: Int = 1, endDim: Int = -1, val name: String = "Flatten") : Module<T, V> (source)

A simple layer that flattens an input tensor into a 1D tensor. This layer has no parameters and simply reshapes the input.

Constructors

constructor(startDim: Int = 1, endDim: Int = -1, name: String = "Flatten")

open override val modules: List<Module<T, V>>

open override val name: String

Human-readable name of the node, typically the module's name.