19 lines
485 B
Python
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),)
|