ConcatAggregationCombination

class ConcatAggregationCombination(aggregation: str | Callable[[Tensor], Tensor] | None = None, dim: int = -1)[来源]

基础类:ConcatCombination

通过连接后沿同一轴进行聚合来组合表示。

初始化组合。

Parameters:

方法总结

forward(xs)

组合一系列单独的表示。

iter_extra_repr()

遍历extra_repr()的组件。

方法文档

forward(xs: Sequence[Tensor]) Tensor[source]

组合一系列单独的表示。

Parameters:

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

Returns:

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

Return type:

Tensor

iter_extra_repr() Iterable[str][来源]

遍历extra_repr()的组件。

此方法通常被重写。一个常见的模式是

def iter_extra_repr(self) -> Iterable[str]:
    yield from super().iter_extra_repr()
    yield "<key1>=<value1>"
    yield "<key2>=<value2>"
Returns:

一个可迭代的extra_repr()的各个组件

Return type:

Iterable[str]