diff --git a/mindspore/ccsrc/common/trans.cc b/mindspore/ccsrc/common/trans.cc
index bc3d5096e1..2853aa9e09 100644
--- a/mindspore/ccsrc/common/trans.cc
+++ b/mindspore/ccsrc/common/trans.cc
@@ -1330,7 +1330,7 @@ bool FracZ3DToNcdhw(const FormatArgs &args, void *result) {
   auto h = args.host_shape[3];
   auto w = args.host_shape[4];
   auto n0 = args.device_shape[1];
-  auto ni = args.device_shape[1];
+  auto ni = args.device_shape[2];
   auto c0 = args.device_shape[3];
   auto hw = h * w;
   auto dhw = d * hw;