plot_forward_pass

plot_forward_pass#

ptyrad.visualization.plot_forward_pass(model, indices, dp_power, show_fig=True, pass_fig=False)[source]#

Plot the forward pass for the input torch model