norm.py

import numpy as np


def norm(x):
    size = x.shape
    res = np.zeros((size[0], size[1]))

    for time_step in range(size[0]):
        for vertex in range(size[1]):
            res[time_step, vertex] = np.linalg.norm(x[time_step, vertex, :])

    return res