zhreshold commented on a change in pull request #11643: Added the diag()
operator
URL: https://github.com/apache/incubator-mxnet/pull/11643#discussion_r201872980
##
File path: tests/python/unittest/test_operator.py
##
@@ -6977,6 +6977,109 @@ def
zhreshold commented on a change in pull request #11643: Added the diag()
operator
URL: https://github.com/apache/incubator-mxnet/pull/11643#discussion_r201871245
##
File path: src/operator/tensor/diag_op-inl.h
##
@@ -0,0 +1,214 @@
+/*
+ * Licensed to the Apache Software