bug fix for fracz_3d to ncdhw trans in host

4 years ago · 83e2a4eb08
parent 2184bcff36
commit 83e2a4eb08
1 changed files with 1 additions and 1 deletions
--- 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;