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.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!
😔 What went wrong?