max_pool3d_forward_f64

void max_pool3d_forward_f64(const double * x, double * y, int * argmax, int B, int C, int D, int H, int W, int KD, int KH, int KW, int OD, int OH, int OW, int sd, int sh, int sw, int pd, int ph, int pw)

source edit

Double-precision counterpart to max_pool3d_forward_f32.

Parameters

xconst double*

Input tensor of shape (B, C, D, H, W).

ydouble*

Output tensor of shape (B, C, OD, OH, OW).

argmaxint32_t*

Per-output winning flat 3-D index.

B, C, D, H, W, KD, KH, KW, OD, OH, OW, sd, sh, sw, pd, ph, pwint

Layout and window parameters; see max_pool3d_forward_f32.