torch_frame.nn.conv.TableConv

class TableConv(*args, **kwargs)[来源]

基础类：Module, ABC

用于表卷积的基类，该卷积按列转换输入的pytorch张量。

abstract forward(x: Tensor, *args: Any, **kwargs: Any) → Any[来源]

将列方向的3维张量处理为另一个列方向的3维张量。

Parameters:

  • x (torch.Tensor) – 输入列方向的张量，形状为 [batch_size, num_cols, hidden_channels]。

  • args (Any) – 额外参数。

  • kwargs (Any) – 额外的关键字参数。

reset_parameters() → None[来源]

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