子模块#

对于 N-Beats 模型的 nn.Modules 的实现。

函数

linear(input_size, output_size[, bias, dropout])

linspace(回溯长度, 预测长度[, ...])

NBEATSBlock(units, thetas_dim[, ...])

初始化内部模块状态,由 nn.Module 和 ScriptModule 共享。

NBEATSGenericBlock(units, thetas_dim[, ...])

初始化内部模块状态,由 nn.Module 和 ScriptModule 共享。

NBEATSSeasonalBlock(units[, thetas_dim, ...])

初始化内部模块状态,由 nn.Module 和 ScriptModule 共享。

NBEATSTrendBlock(units, thetas_dim[, ...])

初始化内部模块状态,由 nn.Module 和 ScriptModule 共享。