mlx.core.broadcast_to

mlx.core.broadcast_to#

broadcast_to(a: scalar | array, /, shape: Sequence[int], *, stream: None | Stream | Device = None) array#

将数组广播到给定的形状。

广播语义与 Numpy 相同。

参数:
  • a (array) – 输入数组。

  • shape (list(int)) – 要广播到的形状。

返回:

具有新形状的输出数组。

返回类型:

array