maximum.py

import numpy as np


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

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

    return res