fig, axs = plt.subplots(4,2, figsize=(16, 24))
for i in range(4):
`im=axs[i,0].imshow(gamma_fn[:,i].reshape(dims), cmap=
