NetReplace

NetReplace[net, patt -> layer] gives a net in which all layers matching patt are replaced with layer.

• NetReplace[net, {rule1, rule2, …}] performs all replacements specified by the rulei.

Examples

net = NetChain[{LinearLayer[10], Tanh, LinearLayer[5]}];
NetReplace[net, Tanh -> Ramp]

Please visit the official Wolfram Language Reference for more details.