NetFlatten

NetFlatten[net] collapses nested NetChain and NetGraph objects within net.

• NetFlatten[net, n] collapses up to nesting level n.

Examples

Flatten a nested network:

NetFlatten[NetChain[{NetChain[{LinearLayer[], Ramp}], LinearLayer[]}]]

Flatten to specific level:

NetFlatten[net, 1]

Please visit the official Wolfram Language Reference for more details.