Skip to content

tensor

Classes

AirTensor

AIR tensor representation.

Attributes

data property writable 
data: NDArray | None

Return the data of the tensor.

size property 
size: int

Return the size of the tensor.

nbytes property 
nbytes: int

Return the number of bytes in the tensor.

is_constant property 
is_constant: bool

Check if the tensor is a constant.

is_persistent property 
is_persistent: bool

Check if the tensor is a persistent variable.

is_scratch property 
is_scratch: bool

Check if the tensor is a scratch tensor.

Functions

expand_dims 
expand_dims(num_dims: int = 4, pre: bool = True)

Expand the tensor dimensions

Parameters:

  • num_dims
    (int, default: 4 ) –

    The target number of dimensions.

  • pre
    (bool, default: True ) –

    If True, expand dimensions at the front; otherwise, at the back.