sub_modules

Implementation of nn.Modules for N-Beats model.

Functions

linear(input_size, output_size[, bias, dropout])
linspace(backcast_length, forecast_length[, ...])

Classes

NBEATSBlock(units, thetas_dim[, ...])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
NBEATSGenericBlock(units, thetas_dim[, ...])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
NBEATSSeasonalBlock(units[, thetas_dim, ...])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
NBEATSTrendBlock(units, thetas_dim[, ...])	Initializes internal Module state, shared by both nn.Module and ScriptModule.