def channels

stelliferous@0.0.2.0 · tensor_band_concat.bend:113

import stelliferous@0.0.2.0/tensor_band_concat.bend as TensorBandConcat
import Base

113def channels(shape: List<&2,U32>) -> U32:
114  match shape:
115    case [channels,height,width]: channels
116    case [batch,channels,height,width]: channels
117    case shape: 0

Uses (2)

Used by (1)