We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e240e75 commit da06daeCopy full SHA for da06dae
array_api_compat/torch/_aliases.py
@@ -711,9 +711,9 @@ def astype(
711
return x.to(dtype=dtype, copy=copy)
712
713
714
-def broadcast_arrays(*arrays: Array) -> list[Array]:
+def broadcast_arrays(*arrays: Array) -> tuple[Array, ...]:
715
shape = torch.broadcast_shapes(*[a.shape for a in arrays])
716
- return [torch.broadcast_to(a, shape) for a in arrays]
+ return tuple(torch.broadcast_to(a, shape) for a in arrays)
717
718
# Note that these named tuples aren't actually part of the standard namespace,
719
# but I don't see any issue with exporting the names here regardless.
0 commit comments