types

Special type annotations used in Tripy.

nvtripy.types.TensorLike

A nvtripy.Tensor or a Python number that can be automatically converted into one.

alias of nvtripy.Tensor | Number

nvtripy.types.IntLike

An integer-like object.

alias of int | nvtripy.DimensionSize

nvtripy.types.ShapeLike

A shape of a nvtripy.Tensor .

alias of Sequence[int | nvtripy.DimensionSize]