cupy.array#

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

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

此函数目前不支持 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()