mindspore.ops.Diag

class mindspore.ops.Diag[源代码]
prim = ops.Diag()
out = prim(input)

等价于

ops.diag(input)

更多详情请查看： mindspore.ops.diag() 。