jittor.init

这里是Jittor的参数初始化模块的API文档,您可以通过from jittor import init来获取该模块。

jittor.init.constant(shape, dtype, value=0.0)[源代码]
jittor.init.constant_(var, value=0.0)[源代码]
jittor.init.gauss(shape, dtype, mean=0.0, std=1.0)[源代码]
jittor.init.gauss_(var, mean=0.0, std=1.0)[源代码]
jittor.init.invariant_uniform(shape, dtype, mode='fan_in')[源代码]
jittor.init.invariant_uniform_(var, mode='fan_in')[源代码]
jittor.init.relu_invariant_gauss(shape, dtype, mode='fan_in')[源代码]
jittor.init.relu_invariant_gauss_(var, mode='fan_in')[源代码]
jittor.init.uniform(shape, dtype, low, high)[源代码]
jittor.init.uniform_(var, low, high)[源代码]