torch.nn.functional.adaptive_max_pool3d_with_indices
function adaptive_max_pool3d_with_indices(input: Tensor, output_size: number | [number, number, number]): PoolWithIndicesResultApplies 3D adaptive max pooling and returns both output and indices. This is equivalent to adaptive_max_pool3d(input, output_size, true).
Parameters
inputTensor- Input tensor of shape (N, C, D, H, W)
output_sizenumber | [number, number, number]- Target output size (D_out, H_out, W_out)
Returns
PoolWithIndicesResult– Tuple of (output, indices) tensors