def bnparams(n):
return cast({'weight': torch.rand(n), 'bias': torch.zeros(n)})
微信
公众号
扫码关注公众号
微信
公众号