I am done

This commit is contained in:
2024-10-30 22:14:35 +01:00
parent 720dc28c09
commit 40e2a747cf
36901 changed files with 5011519 additions and 0 deletions

View File

@ -0,0 +1,17 @@
# Copyright (c) ONNX Project Contributors
# SPDX-License-Identifier: Apache-2.0
from __future__ import annotations
from onnx.helper import np_dtype_to_tensor_dtype
from onnx.reference.ops._op_common_random import _CommonRandom
class Bernoulli(_CommonRandom):
def _run(self, x, dtype=None, seed=None): # type: ignore
if dtype is None:
dtype = np_dtype_to_tensor_dtype(x.dtype)
dtype = self._dtype(x, dtype=dtype, dtype_first=True)
state = self._get_state(seed)
res = state.binomial(1, p=x).astype(dtype)
return (res.astype(dtype),)