NBEATSGenericBlock#

class pytorch_forecasting.models.nbeats.sub_modules.NBEATSGenericBlock(units, thetas_dim, num_block_layers=4, backcast_length=10, forecast_length=5, dropout=0.1)[来源]#

基础: NBEATSBlock

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

方法

forward(x)

定义每次调用时执行的计算。

forward(x)[来源]#

定义每次调用时执行的计算。

应由所有子类重写。

注意

虽然前向传播的公式需要在这个函数内定义,但应该在之后调用 Module 实例,而不是这个,因为前者负责执行注册的钩子,而后者则会默默忽略它们。