num_samples = 10def预测(x): sampled_models = [范围内的指南(无,无))_(范围内)(num_samples)] yhats = [model(x).data for samples in sampled_models] mean = torch.mean(torch.stack(yhats),…
你想要比较 predicted 和 labels 。但是,你的 predicted 是一个 np.array 而 labels 是一个 torch.tensor 因此 eq() (该 == operator)无法比较它们。 更换 np.argmax 同 torch.argmax :
predicted
labels
np.array
torch.tensor
eq()
==
np.argmax
torch.argmax
return torch.argmax(mean, dim=1)
你应该没问题。