[MXNET-1410]Adding Large Tensor Support for tensor transpose

src/operator/tensor/matrix_op-inl.h

@@ -1950,10 +1950,10 @@ struct ReverseParam : public dmlc::Parameter<ReverseParam> {
 #define REVERSE_MAX_DIM 10U
 
 struct reverse {
- MSHADOW_XINLINE static int ReverseIndex(index_t idx,
- index_t nreversedim,
- const index_t * stride_,
- const index_t * trailing_) {
+ MSHADOW_XINLINE static index_t ReverseIndex(index_t idx,
+  index_t nreversedim,
+  const index_t * stride_,
+  const index_t * trailing_) {
  index_t outputIndex = idx;
  for (index_t i = 0; i < nreversedim; ++i) {
  const index_t low = outputIndex % trailing_[i];

tests/nightly/test_large_array.py

@@ -279,6 +279,29 @@ def test_diag():
  assert_almost_equal(r.asnumpy(), np.diag(a_np, k=k))
 
 
+def test_transpose():
+ a = nd.arange(0, LARGE_X).reshape(LARGE_X, 1)
+ b = nd.broadcast_to(a, shape=(a.shape[0], SMALL_Y))
+ t = b.T
+ assert t.shape == (SMALL_Y, LARGE_X)
+ assert np.sum(t[:, -1].asnumpy() == LARGE_X) == b.shape[1]
+
+
+def test_swapaxes():
+ a = nd.arange(0, LARGE_X).reshape(LARGE_X, 1)
+ b = nd.broadcast_to(a, shape=(a.shape[0], SMALL_Y))
+ t = nd.swapaxes(b, dim1=0, dim2=1)
+ assert t.shape == (SMALL_Y, LARGE_X)
+ assert np.sum(t[:, -1].asnumpy() == LARGE_X) == b.shape[1]
+
+def test_flip():
+ a = nd.arange(0, LARGE_X).reshape(LARGE_X, 1)
+ b = nd.broadcast_to(a, shape=(a.shape[0], SMALL_Y))
+ t = nd.flip(b, axis=0)
+ assert t.shape == (LARGE_X, SMALL_Y)
+ assert np.sum(t[-1, :].asnumpy() == 0) == b.shape[1]
+
+
 if __name__ == '__main__':
  import nose
  nose.runmodule()