def geometry

stelliferous@0.0.2.0 · tensor_window.bend:20

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

20def geometry(+shape: List<&2,U32>,+kernel: U32,+stride: U32,+padding: U32) -> Access.Pool:
21  +rank = List.length(&2,U32,shape)
22  +height = Layout.extent_at(shape,Nat.sub(rank,2n))
23  +width = Layout.extent_at(shape,Nat.sub(rank,1n))
24  Access.Pool{height,width,kernel,stride,padding,out_extent(height,kernel,stride,padding),out_extent(width,kernel,stride,padding)}