DPNs
DPNs copied to clipboard
The op of slice_axis?
hi, I have check the json file of your model, and I found that there is a slice-axis op,but I can't find the implementation of this operation in mxnet/src/operator
mxnet/src/operator/matrix_op-inl.h #L843
// slice_axis
MXNET_REGISTER_SIMPLE_OP(slice_axis, XPU)
.set_enable_kwargs(true)
.set_function(XPU::kDevMask, Slice<XPU>,
kNoInplace, kRegisterSymbolic)
.set_gradient(XPU::kDevMask, SliceGrad_<XPU>, kNoInplace)
.set_shape_function(SliceShape)
.describe("Slice the input along certain axis and return a sliced array.")
.add_arguments(SliceParam::__FIELDS__());
got it ! Thanks a lot!