returnn.tensor._dim_extra

Backwards-compatible functions and attribs for the old Dim class, or just rarely used attribs, such that we can save memory for the common case.

class returnn.tensor._dim_extra.DimTypes[source]

Defines possible values for kind.

Unspecified = None[source]
Batch = <batch Entity>[source]
Spatial = <spatial Entity>[source]
Time = <spatial Entity>[source]
Feature = <feature Entity>[source]
Types = (<batch Entity>, <spatial Entity>, <feature Entity>)[source]
class returnn.tensor._dim_extra.Op(kind: str, inputs: List[Dim], attribs: Dict[str, Any] | None = None)[source]

Op on Dim which results in a derived Dim.

Parameters:
  • kind – “add”, “sub”, “mul”, “ceildiv”

  • inputs

  • attribs

returnn.tensor._dim_extra.dim_cmp_value(obj)[source]
Parameters:

obj (Dim|_MarkedDim)

Returns:

anything which can be compared