def pool2d

stelliferous@0.0.2.0 · tensor_window.bend:116

import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors

116def pool2d(~Element: Data,~operation: Element -> Element -> Element,name: String,+pad: Element,tensor: Tensors.Tensor<Element>,
117  +kernel: U32,+stride: U32,+padding: U32) -> Tensors.Tensor<Element>:
118  match tensor:
119    case Tensors.Tensor{storage,shape,fill,status,workers}: pool_status(~Element,~operation,name,status,pad,kernel,stride,padding,storage,shape,fill,workers)
120    case Tensors.Banded{bands,layout,shape,fill,status,workers}: band_status(~Element,~operation,name,status,pad,kernel,stride,padding,layout,bands,shape,fill,workers)