write_CIFAR-10.py

import matplotlib.pyplot as plt
import matplotlib.gridspec as gridspec

# 縦横10枚ずつ画像を描画                                                                                                                                     
def plot(samples):
    fig = plt.figure(figsize=(10, 10))
    gs = gridspec.GridSpec(10, 10)
    gs.update(wspace=0.05, hspace=0.05)

    for i, sample in enumerate(samples):
        ax = plt.subplot(gs[i])
        plt.axis('off')
        ax.set_xticklabels([])
        ax.set_yticklabels([])
        ax.set_aspect('equal')
        plt.imshow(sample.reshape(3,32, 32).transpose(1, 2, 0))

    return fig