Spaces:
Running
Running
# Copyright (c) ONNX Project Contributors | |
# | |
# SPDX-License-Identifier: Apache-2.0 | |
import numpy as np | |
import onnx | |
from onnx.backend.test.case.base import Base | |
from onnx.backend.test.case.node import expect | |
class Acosh(Base): | |
def export() -> None: | |
node = onnx.helper.make_node( | |
"Acosh", | |
inputs=["x"], | |
outputs=["y"], | |
) | |
x = np.array([10, np.e, 1]).astype(np.float32) | |
y = np.arccosh(x) # expected output [2.99322295, 1.65745449, 0.] | |
expect(node, inputs=[x], outputs=[y], name="test_acosh_example") | |
x = np.random.uniform(1.0, 10.0, (3, 4, 5)).astype(np.float32) | |
y = np.arccosh(x) | |
expect(node, inputs=[x], outputs=[y], name="test_acosh") | |