fn
adaptive_max_pool3d_op
→TensorImplPtrint adaptive_max_pool3d_op(const int & x, int OD, int OH, int OW)Python wrappers
Public Python APIs implemented by this engine symbol.Three-dimensional adaptive max-pool to a fixed (OD, OH, OW) volume.
Adaptive variant of max_pool3d_op for video classification and
volumetric backbones that must accept clips of arbitrary duration.
Parameters
xTensorImplPtrInput of shape
(B, C, D, H, W).OD, OH, OWintTarget output extents. All three of
D, H, W must be evenly divisible by the corresponding target.Returns
TensorImplPtrOutput of shape (B, C, OD, OH, OW).
Raises
ShapeMismatchIf
x is not 5-D.not_implementedIf any spatial axis is not evenly divisible by its target.