Module leaf::layers::utility::reshape [] [src]

Utility layer to give a tensor another shape.

This layer should be used as in-place operation, so the tensor that should be reshaped should be specified as both input and output.

Reshaping a tensor is required so that it becomes usable for Layers that interpret meaning into the shape of the tensor.

A lot of layers interpret the last dimensions as NCHW, where the letters stand for:

  • N : number of batch samples
  • C : number of feature maps
  • H : height
  • W : width

Structs

Reshape

Reshape Utility Layer

ReshapeConfig

Specifies configuration parameters for a Reshape Layer.