torch.nn.Module register_buffer()

Register

torch.Tensor into

Torch state_dict but to not appears in learnable parameters

torch Model.parameters(). It is used for registering constant configuration tensor or initialization tensor for the model should be managed by a Tensor in the same device.