pub fn global_max_pool(x: &Tensor, batch: Option<&[usize]>) -> Tensor
Global max pooling for graph-level predictions.