fn
interpolate_trilinear_op
→TensorImplPtrint interpolate_trilinear_op(const int & input, int D_out, int H_out, int W_out, bool align_corners)Public 3-D trilinear resize with autograd.
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.
align_cornersboolCoordinate mapping mode.
Returns
TensorImplPtrTrilinearly resampled tensor of shape (N, C, D_out, H_out, W_out).