pytorch可视化visdom报错could not broa

示例

from visdom import Visdom
vis = Visdom()

data_shot, data_query = batch['xs'], batch['xq']  # [5,5,1,28,28]
vis.images(data_shot)  # 报错

报错

could not broadcast input array from shape (5,1,28,28) into shape (3,1,28)

原因

vis.images文档说明：(数据要求4维)
Given a 4D tensor of shape (B x C x H x W),
or a list of images all of the same size,
makes a grid of images of size (B / nrow, nrow).

解决

from visdom import Visdom
vis = Visdom()

data_shot, data_query = batch['xs'], batch['xq']  # [5,5,1,28,28]
# vis.images(data_shot)  # 报错
for ds in data_shot:
    vis.images(ds)
for dq in data_query:
    vis.images(dq)