maximum.py~

import numpy as np

def maximum(x) :
    size = x.shape 
    res = np.zeros(size[0])

    for time_step in range(size[0]) :
        for vertex in range(size[1]) :
            res[time_step] = max(res[time_step], max(x[time_step, vertex]))

    return res