Module juice::layers::utility::reshape

source ·
Expand description

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§