def reduce_log_sum_exp(unused_ctx, data, axes=None, keepdims=1): if axes is None: axes = range(data.shape.ndims) if not isinstance(axes, (list, tuple)): axes = tuple(axes) return (op.log(op.summation(op.exp(data), axes=axes, keepdims=keepdims)), )
def softplus(data): return (op.log(op.exp(data) + 1.), )
def log(x): return (op.log(x), )
def log(unused_ctx, x): return (op.log(x), )
def softplus(unused_ctx, data): return (op.log(op.exp(data) + 1.), )