CodeWorld

1、predict_result and mask #我们假设每个像素点可能属于3个类别之一：0 or 1 or 2 #注意mask的shape=(1,2,2)，我们要指定batch_size:1，但是不用指定channel,不用对mask做one_hot处理 #四个像素：four piexls mask = torch.tensor([[[1,2],[2,0]]]) #网络的输出shape=（1,3,2,2），batch_size=1,channel=3 pre = torch.tensor(.