def he_uniform(w, a=0, mode='fan_in'):
return nn.kaiming_uniform(w, a=a, mode=mode)
微信
公众号
扫码关注公众号
微信
公众号