Skip to main content

/

Docs

Log in

Mojo module

casting

Ops that modify the shape or data type of a symbolic tensor.

Functions​

  • ​broadcast_to: Broadcasts a symbolic tensor.
  • ​cast: Casts a symbolic tensor to a different data type.
  • ​rebind: Rebinds a symbolic tensor to a specified set of dimensions.
  • ​reshape: Reshapes a symbolic tensor.
  • ​reshape_like: Reshapes a symbolic tensor to the same shape as another symbolic tensor.
  • ​shape_of: Gets the shape of a symbolic tensor as a rank-1 symbolic tensor.
  • ​squeeze: Removes a size-1 dimension from a symbolic tensor.
  • ​transpose: Transposes two dimensions of a symbolic tensor.
  • ​transpose_matrix: Transposes the last two dimensions of a symbolic tensor.
  • ​unsqueeze: Inserts a size-1 dimension into a symbolic tensor.

Was this page helpful?