残差块#

class pytorch_forecasting.models.tide.sub_modules._ResidualBlock(input_dim: int, output_dim: int, hidden_size: int, dropout: float, use_layer_norm: bool)[来源]#

基础: Module

实现TiDE论文中Residual Block的PyTorch模块。

方法

forward(x)

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

forward(x: Tensor) Tensor[来源]#

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

应该被所有子类重写。

注意

尽管前向传播的配方需要在此函数中定义,但之后应该调用 Module 实例而不是这个,因为前者负责运行注册的钩子,而后者则默默忽略它们。