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