def func(node, x): return sum(rm.cross_entropy(node, x, reduce_sum=False))
def func(node, x): return rm.cross_entropy(rm.softmax(node), x)
def func(node, x): return rm.cross_entropy(node, x)