torch_frame.nn.decoder.ExcelFormerDecoder

class ExcelFormerDecoder(in_channels: int, out_channels: int, num_cols: int)

基础类：Decoder

ExcelFormer解码器在“ExcelFormer: A Neural Network Surpassing GBDTs on Tabular Data”论文中介绍。

Parameters:

• in_channels (int) – 输入通道维度

• out_channels (int) – 输出通道的维度

• num_cols (int) – 列数。

reset_parameters() → None

重置模块的所有可学习参数。

forward(x: Tensor) → Tensor

将 x 转换为输出预测。

Parameters:

x (Tensor) – 输入列方向的张量，形状为 [batch_size, num_cols, in_channels]

Returns:

[batch_size, out_channels].

Return type:

张量