ConcatProjectionCombination

class ConcatProjectionCombination(input_dims: ~collections.abc.Sequence[int], output_dim: int | None = None, bias: bool = True, dropout: float = 0.0, activation: str | ~torch.nn.modules.module.Module | type[~torch.nn.modules.module.Module] | None = <class 'torch.nn.modules.linear.Identity'>, activation_kwargs: ~collections.abc.Mapping[str, ~typing.Any] | None = None)[源代码]

基础类:ConcatCombination

通过连接组合表示,然后进行线性投影和激活。

初始化组合。

Parameters:
  • input_dims (Sequence[int]) – 输入维度

  • output_dim (int | None) – 输出维度。默认为第一个输入维度

  • bias (bool) – 是否在线性投影和激活之间添加一个偏置项

  • dropout (float) – 在激活之前使用的dropout

  • activation (str | Module | type[Module] | None) – 激活函数或其提示

  • activation_kwargs (Mapping[str, Any] | None) – 用于实例化激活的额外基于关键字的参数

Raises:

ValueError – 如果 input_dims 为空

方法总结

forward(xs)

组合一系列单独的表示。

方法文档

forward(xs: Sequence[Tensor]) Tensor[来源]

组合一系列单独的表示。

Parameters:

xs (Sequence[Tensor]) – 形状: (*batch_dims, *input_dims_i) 单个表示

Returns:

形状: (*batch_dims, *output_dims) 一个组合表示

Return type:

Tensor