一旦注册,这些值就可以在forward函数中访问,就像其他模块的属性一样。 from torch import nn import torch class ModuleWithCustomValues(nn.Module) ...
確定! 回上一頁