def pad_edge_axes

stelliferous@0.0.2.0 · tensor_layout.bend:509

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

509def pad_edge_axes(~Element: Data,before: List<&2,U32>,after: List<&2,U32>,tensor: Tensors.Tensor<Element>,+axis: U32) -> Tensors.Tensor<Element>:
510  match before after:
511    case Con{+front,before} Con{+back,after}:
512      pad_edge_axes(~Element,before,after,pad_edge_axis(~Element,U32.is_zero(front) && U32.is_zero(back),tensor,axis,front,back),(axis + 1 : U32))
513    case before after: tensor