EnsembleFNN Simple ensemble feed-forward neural network in Pytorch (no for loop) Using torch.einsum to avoid writing for loop over ensemble nets