Mercurial > repos > public > sbplib_julia
diff src/LazyTensors/tuple_manipulation.jl @ 1225:6567e38b05ca refactor/LazyTensors/tuple_manipulation
Replace flatten_tuple with concatenate_tuple
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Sat, 18 Feb 2023 12:24:44 +0100 |
parents | 5bfb182e24dc |
children | ea5b4fca85e0 |
line wrap: on
line diff
--- a/src/LazyTensors/tuple_manipulation.jl Sat Feb 18 12:18:34 2023 +0100 +++ b/src/LazyTensors/tuple_manipulation.jl Sat Feb 18 12:24:44 2023 +0100 @@ -88,19 +88,9 @@ return ntuple(i->cum_szs[i]+1:cum_szs[i+1], length(szs)) end - concatenate_tuples(t::Tuple,ts::Vararg{Tuple}) = (t..., concatenate_tuples(ts...)...) concatenate_tuples(t::Tuple) = t -""" - flatten_tuple(t) - -Takes a nested tuple and flattens the whole structure -""" -flatten_tuple(t::NTuple{N, Number} where N) = t -flatten_tuple(t::Tuple) = ((flatten_tuple.(t)...)...,) # simplify? -flatten_tuple(ts::Vararg) = flatten_tuple(ts) -# TBD: Can concatenate_tuples be used instead? """ left_pad_tuple(t, val, N)