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 Union[nvtripy.Tensor, Number]

nvtripy.types.IntLike

An integer-like object.

alias of Union[int, nvtripy.DimensionSize]

nvtripy.types.ShapeLike

A shape of a nvtripy.Tensor .

alias of Sequence[Union[int, nvtripy.DimensionSize]]