复杂分隔组合

class ComplexSeparatedCombination(combination: str | Combination | type[Combination] | None = None, combination_kwargs: Mapping[str, Any] | None = None, imag_combination: str | Combination | type[Combination] | None = None, imag_combination_kwargs: Mapping[str, Any] | None = None)[source]

基础类:Combination

混合复杂和实数表示的组合。

初始化组合。

注意

如果传递了未实例化的组合,将为实部和虚部分别创建单独的实例

Parameters:
  • 组合 (str | Combination | type[Combination] | None) – 实际的组合,或其提示

  • combination_kwargs (Mapping[str, Any] | None) – 用于实际组合的基于关键字的参数

  • imag_combination (str | Combination | type[Combination] | None) – 虚数组合或其提示。如果为None,则对两者都使用组合。

  • imag_combination_kwargs (Mapping[str, Any] | None) – 用于虚数组合的基于关键字的参数;仅在 imag_combination 不为 None 时使用

方法总结

forward(xs)

组合一系列单独的表示。

output_shape(input_shapes)

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

方法文档

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, ...][source]

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

注意

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

Parameters:

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

Returns:

输出形状

Return type:

tuple[int, …]