DimensionSize

class nvtripy.DimensionSize(data: int, name: str | None = None)[source]

Bases: Tensor

A 0D, int32 tensor that represents a scalar value extracted from the shape of a tensor.

Parameters:

  • data (int) – The value of the DimensionSize, which should be a scalar integer.

  • name (str | None) – An optional name.