fn
interpolate_nearest_3d_op
→TensorImplPtrint interpolate_nearest_3d_op(const int & input, int D_out, int H_out, int W_out)Public 3-D nearest-neighbour resize (no autograd).
Volumetric counterpart of interpolate_nearest_2d_op; produces a
(N, C, D_out, H_out, W_out) output by floor-rounding source voxel
coordinates. No backward node is attached.
Parameters
inputTensorImplPtr5-D input tensor
(N, C, D_in, H_in, W_in).D_outintDesired output depth.
H_outintDesired output height.
W_outintDesired output width.
Returns
TensorImplPtrResampled tensor of shape (N, C, D_out, H_out, W_out).