def output_shape

stelliferous@0.0.2.0 · tensor_convolution.bend:51

import stelliferous@0.0.2.0/tensor_convolution.bend as TensorConvolution

import Base

51def output_shape(batched: Bool,batch: U32,rows: U32,height: U32,width: U32) -> List<&2,U32>:
52  match batched:
53    case False{}: [rows,height,width]
54    case True{}: [batch,rows,height,width]

Uses (3)

Used by (2)