Skip to main content

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.