98import torch 99from torch import nn 100 101from labml_helpers.module import Module ... 160 self.register_buffer('exp_var', torch.ones(channels)).
確定! 回上一頁