label.shape=(n, 28, 28)
label = torch.repeat_interleave(label.unsqueeze(dim=1), repeats=5, dim=1) # (n, 8, 56, 56)
label.shape=(n, 5, 28, 28)
torch——通过复制扩展维度
label.shape=(n, 28, 28)
label = torch.repeat_interleave(label.unsqueeze(dim=1), repeats=5, dim=1) # (n, 8, 56, 56)
label.shape=(n, 5, 28, 28)
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
扫一扫
