code: def plot_matrix(ax, r, g, b, t, render=False):
      if ax:
          plt.cla()
          ax.imshow(
              np.concatenate(
                  (
                      r[..., np.newaxis],
                      g[..., np.newaxis],
                      b[..., np.newaxis],
                  ),
                  axis=2,
              )
          )
          ax.set_xticks([])
          ax.set_yticks([])
          plt.draw()
          if render:
              plt.savefig(
                  "step{0:03d}.png".format(t), bbox_inches="tight"
              )
          plt.pause(0.001)
  
file_path: autograd/examples/fluidsim/wing.py
imports: from __future__ import absolute_import
  from __future__ import print_function
  from builtins import range
  import autograd.numpy
  from autograd import value_and_grad
  from scipy.optimize import minimize
  import matplotlib.pyplot
  import os
label_desc: Use image node_id 7 for calling a global function with example usage: plot_matrix(ax, r, g, b, t, render) without return types