Function init_

model/modeling_bert.py:43–44

(tensor)

def normal_init_method(mean, std):
    def init_(tensor):
        return torch.nn.init.normal_(tensor, mean=mean, std=std)

    return init_