def METHOD_NAME(self, axis):
    x = tensor4("x")
    rng = np.random.default_rng(utt.fetch_seed())
    xv = rng.standard_normal((2, 3, 4, 5)).astype(config.floatX)
    f = function([x], softmax(x, axis=axis))
    assert np.allclose(f(xv), scipy_softmax(xv, axis=axis))