门控残差网络#
- class pytorch_forecasting.models.temporal_fusion_transformer.sub_modules.GatedResidualNetwork(input_size: int, hidden_size: int, output_size: int, dropout: float = 0.1, context_size: int = None, residual: bool = False)[来源]#
基础:
Module初始化内部模块状态,由 nn.Module 和 ScriptModule 共享。
方法
forward(x[, context, residual])定义每次调用时执行的计算。
init_weights()