组合模块

class CompositionModule(*args, **kwargs)[source]

基础类：Module, ABC

向量的（逐元素）组合函数。

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

方法总结

forward(a, b)

组合两批向量。

方法文档

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

组合两批向量。

张量必须是可广播的。

Parameters:

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

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

Returns:

形状: s

Return type:

Tensor