cuquantum.cutensornet.create_tensor_descriptor

cuquantum.cutensornet.create_tensor_descriptor(intptr_t handle, int32_t num_modes, extents, strides, modes, int data_type) → intptr_t

初始化 cutensornetTensorDescriptor_t，描述张量的信息。

参数

• handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。

• num_modes (int32_t) – 张量的模式数量。

• extents (object) –

  大小为 num_modes 的数组；extents[j] 对应于张量第 j 个模式的范围。它可以是

  • 一个 int 作为数组的指针地址，或者

  • 一个 int64_t 的 Python 序列。

• strides (object) –

  大小为 num_modes 的数组；strides[j] 对应于相对于张量第 j 个模式的两个逻辑相邻元素之间的线性偏移量（以物理内存为单位）。它可以是

  • 一个 int 作为数组的指针地址，或者

  • 一个 int64_t 的 Python 序列。

• modes (object) –

  大小为 num_modes 的数组；modes[j] 表示张量的第 j 个模式。它可以是

  • 一个 int 作为数组的指针地址，或者

  • 一个 int32_t 的 Python 序列。

• data_type (int) – 表示张量的数据类型。

返回

指向 cutensornetTensorDescriptor_t 的指针。

返回类型

intptr_t

参见

cutensornetCreateTensorDescriptor()