cupy.array#

cupy.array(obj, dtype=None, copy=True, order='K', subok=False, ndmin=0, *, blocking=False)[source]#

在当前设备上创建一个数组。

此函数目前不支持 subok 选项。

参数:
  • objcupy.ndarray 对象或可以传递给 numpy.array() 的任何其他对象。

  • dtype – 数据类型说明符。

  • copy (bool) – 如果为 False,此函数将尽可能返回 obj。否则,此函数始终返回一个新数组。

  • order ({'C', 'F', 'A', 'K'}) – 行主序 (C-风格) 或列主序 (Fortran-风格)。当 order'A' 时,如果 a 是列主序,则使用 'F',否则使用 'C'。当 order'K' 时,它尽可能保持步幅。如果 objnumpy.ndarray,函数返回 'C''F' 顺序数组。

  • subok (bool) – 如果为 True,则子类将被传递,否则返回的数组将被强制为基类数组(默认)。

  • ndmin (int) – 最小维度数。如果需要,将在形状的头部插入一维。

  • blocking (bool) – 默认为 False,意味着如果需要主机到设备 (H2D) 复制,它将在当前流上异步运行,用户负责确保流的顺序。例如,在复制时没有正确顺序地写入源 obj 将导致竞态条件。如果设置为 True,复制是同步的(相对于主机)。

返回:

当前设备上的一个数组。

返回类型:

cupy.ndarray

注意

此方法目前不支持 subok 参数。

注意

如果 obj 是包含大端字节序数据的 numpy.ndarray 实例,此函数会自动将其字节序转换为小端字节序,这是 NVIDIA 和 AMD GPU 架构的原生用法。

另请参阅

numpy.array()