DPNs icon indicating copy to clipboard operation
DPNs copied to clipboard

The op of slice_axis?

Open chenfsjz opened this issue 7 years ago • 2 comments

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

chenfsjz avatar Jul 19 '17 02:07 chenfsjz

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__());

cypw avatar Jul 19 '17 02:07 cypw

got it ! Thanks a lot!

chenfsjz avatar Jul 19 '17 03:07 chenfsjz