提交 ef5592ff authored 作者: ricardoV94's avatar ricardoV94 提交者: Ricardo Vieira

Add docstrings for `combine_dims_and_shape`

上级 2d9a7aeb
...@@ -23,7 +23,12 @@ def combine_dims_and_shape( ...@@ -23,7 +23,12 @@ def combine_dims_and_shape(
) -> dict[str, int | None]: ) -> dict[str, int | None]:
"""Combine information of static dimensions and shapes from multiple xtensor inputs. """Combine information of static dimensions and shapes from multiple xtensor inputs.
Exclude Parameters
----------
inputs: Sequence of XTensorVariable
Sequence of input XTensorVariables.
exclude: Sequence of str, optional
Dimensions to exclude from the result.
""" """
exclude_set: set[str] = set() if exclude is None else set(exclude) exclude_set: set[str] = set() if exclude is None else set(exclude)
dims_and_shape: dict[str, int | None] = {} dims_and_shape: dict[str, int | None] = {}
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论