Dear Thomas
Thank you for your explanation
But how should I modify the source above?
Should I write down like this??
output = features.new(num_rois, num_channels, ctx.aligned_height, ctx.width).zero_()