Implement torch diagflat, and movedim as well as it is also needed as a helper function (#7378)

Implement torch diagflat, and movedim as well as it is also needed as a helper function (#7378) #664