src/arraymancer/nn/layers/flatten

Imports

proc forward[T](self: Flatten[T]; input: Variable[Tensor[T]]): Variable[ Tensor[T]]: Source Edit
proc init[T](ctx: Context[Tensor[T]]; layerType: typedesc[Flatten[T]]; inShape: seq[int]): Flatten[T]: Creates a flattening layer, which "flattens" its input by reshaping it into a one-dimensional tensor. inShape describes the expected shape of the input. Source Edit
func inShape[T](self: Flatten[T]): seq[int]: Source Edit
func outShape[T](self: Flatten[T]): seq[int]: Source Edit