ReshapeLayer

ReshapeLayer[spec] represents a net layer that reshapes the input array according to the specification spec.

Examples

ReshapeLayer[{2, 3}]

ReshapeLayer[{4, 4}][RandomReal[1, {2, 8}]]

Please visit the official Wolfram Language Reference for more details.