mindspore.ops.Diag

class mindspore.ops.Diag[source]
prim = ops.Diag()
out = prim(input)

is equivalent to

ops.diag(input)

Refer to mindspore.ops.diag() for more details.