功能组合模块

class FunctionalCompositionModule(*args, **kwargs)[源代码]

基础：CompositionModule

通过函数进行组合（即无状态）。

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

方法总结

forward(a, b)

组合两批向量。

方法文档

forward(a: Tensor, b: Tensor) → Tensor[来源]

组合两批向量。

张量必须是可广播的。

Parameters:

  • a (Tensor) – 形状: s_1 第一个张量。

  • b (Tensor) – 形状: s_2 第二个张量。

Returns:

形状: s

Return type:

Tensor