Visualize images from loop using matplotlib subplot

viz.py

import matplotlib.pyplot as plt

fig = plt.figure(figsize= (10, 10))
for i in range (0,12):
  # required nrows=4, required ncoms=4, index_location= i+1
  ax = fig.add_subplot(4, 4, i+1)
  # x_batch[i]: Image object at each iteration
  ax.imshow(x_batch[i])