    def __init__(self, ord_kind=2, dim=None, keepdim=False, dtype=None) -> None:
        """initialize the op"""
        if dim is None:
            raise NotImplementedError(
                "flattening input tensor before normalization is not supported yet"
            )
        super().__init__(dim, keepdim, dtype)
        self._attrs["op"] = "vector_norm"
        self._attrs["ord_kind"] = str(ord_kind)
initialize the op
python/aitemplate/compiler/ops/reduce/vector_norm.py