torchvision.utils.make_gridの結合の順番を確認する

なにこれ

• torchvision.utils.make_gridで結合したテンソルが、縦横どちらに向かって結合されていくかを確認する
• 結論：Zの字の方向に結合する。

実行するコード

• save_imageはmake_gridを呼ぶ仕様なので、save_imageに直接tensorを渡す
• 少しずつ濃くなるタイルを生成して、順番に並べる

import torch
from torchvision.utils import save_image
imgs = list()
for i in range(9):
    imgs.append(torch.ones([1, 64, 64])*i/10)
imgs = torch.stack(imgs)
save_image(imgs, "grid.png", 3)

出力

結論

• 左上から、行方向に進む
• Zの字の方向に結合する。