Mojo function
rebind
rebind(v: Symbol, out_dims: List[Dim, 0], 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, 0]
): The symbolic shape to assert forv
, as a list ofDim
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?