mlx.core.as_strided

目录

mlx.core.as_strided#

as_strided(a: array, /, shape: Sequence[int] | None = None, strides: Sequence[int] | None = None, offset: int = 0, *, stream: None | Stream | Device = None) → array#

使用给定的形状和步长创建数组的视图。

无论提供的数组的存储顺序和当前步长如何，结果数组始终会像提供的数组是行主序一样。

注意

请注意，此函数应谨慎使用，因为它直接改变数组的形状和步长。这可能导致结果数组指向无效的内存位置，从而导致程序崩溃。

参数：

  • a (array) – 输入数组

  • shape (list(int), 可选) – 结果数组的形状。如果为 None，则默认为 a.shape()。

  • strides (list(int), 可选) – 结果数组的步长。如果为 None，则默认为 a.shape() 的反向累积乘积（不包含当前元素）。

  • offset (int) – 从输入数组的开头跳过这么多元素。

返回值：

输出数组，它是输入的步长视图。

返回类型：

array

目录