pub fn expand<T: CubePrimitive, K: TensorMapKind>(
scope: &Scope,
tensor_map: &<TensorMap<T, K> as CubeType>::ExpandType,
) -> <u64 as CubeType>::ExpandTypepub fn expand<T: CubePrimitive, K: TensorMapKind>(
scope: &Scope,
tensor_map: &<TensorMap<T, K> as CubeType>::ExpandType,
) -> <u64 as CubeType>::ExpandType