code: public INDArray output(INDArray input, TrainingMode train) {
          return output(input, train == TrainingMode.TRAIN);
      }
docs: Perform inference on the provided input/features - i.e., perform forward pass using the provided input/features
  and return the output of the final layer.
  
  @param input Input to the network
  @param train whether the output is test or train. This mainly affect hyper parameters such as dropout and
  batch normalization, which have different behaviour for test vs. train
  @return The network predictions - i.e., the activations of the final layer