... and input is your input tensor predictions = [m[i].fit(input) for i in models] avg_prediction = torch.stack(b).mean(0).argmax() The stack method ...
確定! 回上一頁