module
casting
Ops that modify the shape or data type of a symbolic tensor.
Functions
-
shape_of
: Gets the shape of a symbolic tensor as a rank-1 symbolic tensor. -
cast
: Casts a symbolic tensor to a different data type. -
rebind
: Rebinds a symbolic tensor to a specified set of dimensions. -
squeeze
: Removes a size-1 dimension from a symbolic tensor. -
unsqueeze
: Inserts a size-1 dimension into a symbolic tensor. -
reshape
: Reshapes a symbolic tensor. -
reshape_like
: Reshapes a symbolic tensor to the same shape as another symbolic tensor. -
transpose
: Transposes two dimensions of a symbolic tensor. -
transpose_matrix
: Transposes the last two dimensions of a symbolic tensor.