Source code for onnx_diagnostic.reference.ops.op_mul_sigmoid

import numpy as np
from onnx.reference.op_run import OpRun


def sigmoid(x):  # type: ignore
    if x > 0:
        return 1 / (1 + np.exp(-x))
    return np.exp(x) / (1 + np.exp(x))




class MulSigmoid(OpRun):
    op_domain = "onnx_extended.ortops.optim.cuda"

    def __init__(self, onnx_node, run_params):  # type: ignore
        OpRun.__init__(self, onnx_node, run_params)
        self.vf = np.vectorize(sigmoid)

    def _run(self, X):
        if len(X.shape) == 0:
            return ((X * sigmoid(X)).astype(X.dtype),)
        if X.size == 0:
            return (X,)
        return ((X * self.vf(X)).astype(X.dtype),)