Files
2024-10-30 22:14:35 +01:00

19 lines
485 B
Python

# Copyright (c) ONNX Project Contributors
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations
import numpy as np
from onnx.reference.op_run import OpRun
def common_reference_implementation(data: np.ndarray, shape: np.ndarray) -> np.ndarray:
ones = np.ones(shape, dtype=data.dtype)
return data * ones # type: ignore
class Expand(OpRun):
def _run(self, data, shape): # type: ignore
return (common_reference_implementation(data, shape),)