def middle_shape

stelliferous@0.0.2.0 · tensor_window.bend:49

import Base
import stelliferous@0.0.2.0/tensor_access.bend as Access

49# The column-swept planes keep the rows of the input and have the columns of
50# the output.
51def middle_shape(+shape: List<&2,U32>,pool: Access.Pool) -> List<&2,U32>:
52  Access.Pool{height,width,kernel,stride,padding,out_height,out_width} = pool
53  List.append(&2,U32,List.take(&2,U32,shape,Nat.sub(List.length(&2,U32,shape),2n)),[height,out_width])