With register_buffer, we can make sure that the new tensor we have created will be part of state_dict. The mask tensor is then filled with 1s using the ...
確定! 回上一頁