Computes the volume of an iterable.

Parameters:iterable – Any python iterable, including a Dims object.
Returns:The volume of the iterable (0 for empty iterables).


class tensorrt.Dims(*args, **kwargs)

Structure to define the dimensions of a tensor. Dims and all derived classes behave like Python tuple s. Furthermore, the TensorRT API can implicitly convert Python iterables to Dims objects, so tuple or list can be used in place of this class.

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.Dims) -> None
  2. __init__(self: tensorrt.tensorrt.Dims, shape: List[int]) -> None

The maximum number of dimensions supported by Dims.

get_type(self: tensorrt.tensorrt.Dims, arg0: int) → tensorrt.tensorrt.DimensionType

Queries the type of a dimension.

Returns:The type of the specified dimension.


class tensorrt.Dims2(*args, **kwargs)

Structure to define 2D shape.

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.Dims2) -> None
  2. __init__(self: tensorrt.tensorrt.Dims2, dim0: int, dim1: int) -> None
  3. __init__(self: tensorrt.tensorrt.Dims2, shape: List[int]) -> None


class tensorrt.DimsHW(*args, **kwargs)

Structure to define 2D shape with height and width.

  • hint The first dimension (height).
  • wint The second dimension (width).

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.DimsHW) -> None
  2. __init__(self: tensorrt.tensorrt.DimsHW, h: int, w: int) -> None
  3. __init__(self: tensorrt.tensorrt.DimsHW, shape: List[int]) -> None


class tensorrt.Dims3(*args, **kwargs)

Structure to define 3D shape.

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.Dims3) -> None
  2. __init__(self: tensorrt.tensorrt.Dims3, dim0: int, dim1: int, dim2: int) -> None
  3. __init__(self: tensorrt.tensorrt.Dims3, shape: List[int]) -> None


class tensorrt.DimsCHW(*args, **kwargs)

Structure to define 3D tensor with a channel dimension, height, and width.

  • cint The first dimension (channel).
  • hint The second dimension (height).
  • wint The third dimension (width).

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.DimsCHW) -> None
  2. __init__(self: tensorrt.tensorrt.DimsCHW, c: int, h: int, w: int) -> None
  3. __init__(self: tensorrt.tensorrt.DimsCHW, shape: List[int]) -> None


class tensorrt.Dims4(*args, **kwargs)

Structure to define 4D tensor.

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.Dims4) -> None
  2. __init__(self: tensorrt.tensorrt.Dims4, dim0: int, dim1: int, dim2: int, dim3: int) -> None
  3. __init__(self: tensorrt.tensorrt.Dims4, shape: List[int]) -> None


class tensorrt.DimsNCHW(*args, **kwargs)

Structure to define 4D tensor with a batch dimension, a channel dimension, height and width.

  • nint The first dimension (batch).
  • cint The second dimension (channel).
  • hint The third dimension (height).
  • wint The fourth dimension (width).

Overloaded function.

  1. __init__(self: tensorrt.tensorrt.DimsNCHW) -> None
  2. __init__(self: tensorrt.tensorrt.DimsNCHW, n: int, c: int, h: int, w: int) -> None
  3. __init__(self: tensorrt.tensorrt.DimsNCHW, shape: List[int]) -> None