Fix op transpose_copy (#758)

Summary:
Pull Request resolved: #758

ghstack-source-id: 203458980
exported-using-ghexport

Reviewed By: SS-JIA

Differential Revision: D50104694

fbshipit-source-id: e1020aeca472281e77c45b2e144410ed4d64a4cf

Author: manuelcandales

kernels/portable/cpu/op_transpose_copy.cpp

@@ -39,10 +39,10 @@ Tensor& transpose_copy_int_out(
       out);
 
   if (dim0 < 0) {
-    dim0 += nonzero_dim(out);
+    dim0 += nonzero_dim(in);
   }
   if (dim1 < 0) {
-    dim1 += nonzero_dim(out);
+    dim1 += nonzero_dim(in);
   }
 
   Tensor::SizesType expected_out_size[kTensorDimensionLimit];

kernels/portable/cpu/util/transpose_util.h

@@ -160,8 +160,11 @@ inline void get_transpose_out_target_size(
     size_t* out_ndim) {
   *out_ndim = in.dim();
 
-  size_t i = 0;
-  for (; i < in.dim() - 1; ++i) {
+  if (in.dim() == 0) {
+    return;
+  }
+
+  for (size_t i = 0; i < in.dim(); ++i) {
     out_sizes[i] = in.size(i);
   }
   out_sizes[dim0] = in.size(dim1);