code: def make_pinwheel(
      radial_std,
      tangential_std,
      num_classes,
      num_per_class,
      rate,
      rs=npr.RandomState(0),
  ):
      """Based on code by Ryan P. Adams."""
      rads = np.linspace(0, 2 * np.pi, num_classes, endpoint=False)
  
      features = rs.randn(num_classes * num_per_class, 2) * np.array(
          [radial_std, tangential_std]
      )
      features[:, 0] += 1
      labels = np.repeat(np.arange(num_classes), num_per_class)
  
      angles = rads[labels] + rate * np.exp(features[:, 0])
      rotations = np.stack(
          [
              np.cos(angles),
              -np.sin(angles),
              np.sin(angles),
              np.cos(angles),
          ]
      )
      rotations = np.reshape(rotations.T, (-1, 2, 2))
  
      return np.einsum("ti,tij->tj", features, rotations)
  
file_path: autograd/examples/data.py
imports: from __future__ import absolute_import
  import matplotlib.pyplot
  import matplotlib.image
  import autograd.numpy
  import autograd.numpy.random
  import data_mnist
label_desc: Use image node_id 4 for calling a global function with example usage: make_pinwheel(radial_std, tangential_std, num_classes, num_per_class, rate, rs) and returns: np