`NDArray.max()` should support max from multiple dimensions. However, the current version `0.21.0` of Tensorflow.NET only supports `max` on one dimension.