Mojo function

rebind

rebind(v: Symbol, out_dims: List[Dim], message: String) -> Symbol

Rebinds a symbolic tensor to a specified set of dimensions.

This does not mutate the symbolic tensor passed in, but instead adds a runtime assert that the input symbolic shape is equivalent to out_dims shape. For example, if the input tensor shape has dynamic/unknown sizes, this will assert a fixed sizes that may be required for a subsequent operation.

Args:

• ​v (Symbol): The input symbolic tensor to rebind.
• ​out_dims (List[Dim]): The symbolic shape to assert for v, as a list of Dim values.
• ​message (String): The message printed if the rebind fails at runtime.

Returns:

A symbolic tensor with the same elements and shape as the given tensor, but with the symbolic shape asserted to out_dims.

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit