cupy.diag#
- cupy.diag(v, k=0)[源代码][源代码]#
返回一个对角线或一个对角线数组。
- 参数:
v (array-like) – 数组或类似数组的对象。
k (int) – 对角线索引。零表示主对角线，正值表示上对角线，负值表示下对角线。
- 返回:
如果
v表示一个一维数组，则返回一个二维数组，其中指定对角线由v填充。如果v表示一个二维数组，则返回v的指定对角线。在后一种情况下，如果v是一个cupy.ndarray对象，则返回其视图。- 返回类型:
参见
numpy.diag()