Hi, thank you for the nice package! When I am looking into the PlanarLayer and RadialLayer, is it worth considering adding rng like:
function PlanarLayer(dims::Int, wrapper=identity, rng=Random.GLOBAL_RNG)
w = wrapper(randn(rng, dims))
u = wrapper(randn(rng, dims))
b = wrapper(randn(rng, 1))
return PlanarLayer(w, u, b)
end
function RadialLayer(dims::Int, wrapper=identity, rng=Random.GLOBAL_RNG)
α_ = wrapper(randn(rng, 1))
β = wrapper(randn(rng, 1))
z_0 = wrapper(randn(rng, dims))
return RadialLayer(α_, β, z_0)
end
Hi, thank you for the nice package! When I am looking into the
PlanarLayer
andRadialLayer
, is it worth considering addingrng
like: