~/jendhub

def max_pool2d_with

stelliferous@0.0.2.0 · tensor_f32.bend:350

import stelliferous@0.0.2.0/tensor_f32.bend as TensorF32
import Base
import stelliferous@0.0.2.0/tensor_window.bend as Window

350# Over the last two axes; positions outside the input count as -inf.
351def max_pool2d_with(tensor: Tensor(),+kernel: U32,+stride: U32,+padding: U32) -> Tensor():
352  Window.pool2d(~F32,~F32.max,"max_pool2d",F32.div(F32.neg(1.0),0.0),tensor,kernel,stride,padding)

Uses (7)

Used by (1)