DimensionSize

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

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.

eval() → DimensionSize

Immediately evaluates this DimensionSize object.

Note

DimensionSize will always reside on host even after it is evaluated.

Returns:

The evaluated DimensionSize.

Return type:

DimensionSize

Example

dim_size = tp.ones((2, 2)).shape[0]
dim_size.eval()
print(dim_size.device)

Local Variables

>>> dim_size
2

Output

cpu:0

tolist() → List | Number

Returns the tensor as a nested list. If the tensor is a scalar, returns a python number.

Returns:

The tensor represented as a nested list or a python number.

Return type:

List | Number

Example: Ranked tensor

tensor = tp.ones((2, 2))
tensor_list = tensor.tolist()

Local Variables

>>> tensor_list
[[1.0, 1.0], [1.0, 1.0]]

Example: Scalar

tensor = tp.Tensor(2.0)
tensor_scalar = tensor.tolist()

Local Variables

>>> tensor_scalar
2.0