Search

nn.Module.register_buffer(name, tensor, persistent=True)

self.register_buffer("relative_position_index", relative_position_index)
nn.Module 안에서 위와 같이 사용한다.
register_parameter와 달리 단순 buffer의 역할로,
optimizer로 (backpropagation으로) update되지 않는다.
state_dict에 저장된다 ( 다시 사용 가능 )
GPU에서 작동한다.