组合

class Combination(*args, **kwargs)

基础类: Module, ExtraReprMixin, ABC

组合的基类。

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

方法总结

forward(xs)	组合一系列单独的表示。
output_shape(input_shapes)	计算给定输入形状的输出形状。

方法文档

abstract forward(xs: Sequence[Tensor]) → Tensor

组合一系列单独的表示。

Parameters:

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

Returns:

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

Return type:

Tensor

output_shape(input_shapes: Sequence[tuple[int, ...]]) → tuple[int, ...]

计算给定输入形状的输出形状。

注意

如果没有可用的符号计算，此方法将运行一次前向传递。

Parameters:

input_shapes (Sequence[tuple[int, ...]]) – 输入形状，不包括批次维度

Returns:

输出形状

Return type:

tuple[int, …]