Skip to main content

Mojo function

shape_of

shape_of(v: Symbol) -> Symbol

Gets the shape of a symbolic tensor as a rank-1 symbolic tensor.

Args:

  • v (Symbol): The symbolic tensor whose shape is returned.

Returns:

A symbolic rank-1 tensor representing the input's shape.