mxnet尝试进行除法计算失败

harry_zhu

library(mxnet)
library(dplyr)

supply_tensor = mxnet::mx.nd.array(
  array(
    rep(
      c(c(c(1,2,3,4,5),
          c(1,2,3,4,5),
          c(1,2,3,4,5),
          c(1,2,3,4,5),
          c(1,2,3,4,5))
      ),4)
    ,dim=c(5,5,4,1)
  ),ctx = mxnet::mx.cpu()
)


demand_tensor = mxnet::mx.nd.array(
  array(
    seq(1,100),dim=c(5,5,4,1))
  ,ctx = mxnet::mx.cpu()
)

sum_supply_tensor = mxnet::mx.nd.sum.axis(data = supply_tensor,axis = c(2,3))
supply_tensor %>% 
    mxnet::mx.nd.broadcast.div(
       sum_supply_tensor %>% 
       mxnet::mx.nd.expand.dims(1) %>% 
       mxnet::mx.nd.expand.dims(1)
)

harry_zhu

错误信息

[12:37:32] /Users/kouqiang/Documents/mxnet/dmlc-core/include/dmlc/logging.h:304: [12:37:32] src/operator/tensor/./elemwise_binary_broadcast_op.h:48: Check failed: l == 1 || r == 1 operands could not be broadcast together with shapes (1,4,5,5) (1,1,1,4)

Stack trace returned 10 entries:
[bt] (0) 0   libmxnet.so                         0x0000000112be69c8 _ZN4dmlc15LogMessageFatalD2Ev + 40
[bt] (1) 1   libmxnet.so                         0x0000000112de00f0 _ZN5mxnet2op20BinaryBroadcastShapeERKN4nnvm9NodeAttrsEPNSt3__16vectorINS1_6TShapeENS5_9allocatorIS7_EEEESB_ + 1392
[bt] (2) 2   libmxnet.so                         0x00000001131b9c7c _Z12SetShapeTypePKN4nnvm2OpERKNS_9NodeAttrsERKN5mxnet7ContextERKNSt3__16vectorINS6_7NDArrayENSA_9allocatorISC_EEEERKiPSF_ + 1276
[bt] (3) 3   libmxnet.so                         0x00000001131bdfbb _Z20ImperativeInvokeImplRKN4nnvm9NodeAttrsEiPPvPiPS4_ + 955
[bt] (4) 4   libmxnet.so                         0x00000001131bf60b MXImperativeInvoke + 171
[bt] (5) 5   mxnet.so                            0x0000000113a24f72 _ZN5mxnet1R15NDArrayFunctionclEPP7SEXPREC + 1314
[bt] (6) 6   Rcpp.so                             0x00000001100d8e71 _Z23InternalFunction_invokeP7SEXPREC + 177
[bt] (7) 7   libR.dylib                          0x000000010cdb458d do_External + 381
[bt] (8) 8   libR.dylib                          0x000000010cde5b3d Rf_eval + 1661
[bt] (9) 9   libR.dylib                          0x000000010cdf7f80 do_begin + 608


 *** caught segfault ***
address 0x0, cause 'memory not mapped'

Traceback:
 1: .External(list(name = "InternalFunction_invoke", address = <pointer: 0x7f8818736a90>,     dll = list(name = "Rcpp", path = "/Library/Frameworks/R.framework/Versions/3.4/Resources/library/Rcpp/libs/Rcpp.so",         dynamicLookup = TRUE, handle = <pointer: 0x7f8818650c50>,         info = <pointer: 0x7f881981e440>), numParameters = -1L),     <pointer: 0x7f881d063f10>, lhs, rhs, out)
 2: mxnet::mx.nd.broadcast.div(., sum_supply_tensor %>% mxnet::mx.nd.expand.dims(1) %>%     mxnet::mx.nd.expand.dims(1))
 3: function_list[[k]](value)
 4: withVisible(function_list[[k]](value))
 5: freduce(value, `_function_list`)
 6: `_fseq`(`_lhs`)
 7: eval(quote(`_fseq`(`_lhs`)), env, env)
 8: eval(quote(`_fseq`(`_lhs`)), env, env)
 9: withVisible(eval(quote(`_fseq`(`_lhs`)), env, env))
10: supply_tensor %>% mxnet::mx.nd.broadcast.div(sum_supply_tensor %>%     mxnet::mx.nd.expand.dims(1) %>% mxnet::mx.nd.expand.dims(1))