5cf0a414aa1f7c17fdc0eced712de551c335a25c,src/graph_transpiler/webdnn/frontend/chainer/functions/activation.py,,_convert_log_softmax,#Any#Any#,65

Before Change

@ChainerConverter.register_handler("LogSoftmax")
def _convert_log_softmax(converter: ChainerConverter, c_op: "chainer.functions.LogSoftmax"):
    // TODO
    raise NotImplementedError("[ChainerConverter] LogSoftmax is not supported")


// noinspection PyUnusedLocal
@ChainerConverter.register_handler("LSTM")

After Change

@ChainerConverter.register_handler("LogSoftmax")
def _convert_log_softmax(converter: ChainerConverter, c_op: "chainer.functions.LogSoftmax"):
    x = converter.get_variable(c_op.inputs[0])
    axis = x.order.axes[1]

    max_x, = Max(None, axis=axis)(x)
    exp_delta_x, = Exp(None)(x - max_x)
    sum_exp_delta_x, = Sum(None, axis=axis)(exp_delta_x)
    log_sum_delta_exp, = Log(None)(sum_exp_delta_x)

    y = x - (log_sum_delta_exp + max_x)
    converter.set_variable(c_op.outputs[0](), y)


// noinspection PyUnusedLocal

Instances