Table of Contents
Shortcuts

thunder.clang.maybe_broadcast

thunder.clang.maybe_broadcast(*args, treat_cpu_scalar_tensors_as_numbers=True, expand_fn)[source]

Returns tensors with the same shape, possibly broadcasting inputs to the result shape.

Parameters:

expand_fn (Callable) –

To analyze traffic and optimize your experience, we serve cookies on this site. By clicking or navigating, you agree to allow our usage of cookies. Read PyTorch Lightning's Privacy Policy.