Mercurial > repos > public > sbplib_julia
diff src/Grids/tensor_grid.jl @ 1438:66ad232e45a1 refactor/grids/tensor_grid_eltype
Merge default
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Wed, 22 Nov 2023 17:47:32 +0100 |
| parents | 25af92b4a7ea e3a80ef08d09 |
| children | c0c5e0620b83 |
line wrap: on
line diff
--- a/src/Grids/tensor_grid.jl Thu Sep 07 16:41:49 2023 +0200 +++ b/src/Grids/tensor_grid.jl Wed Nov 22 17:47:32 2023 +0100 @@ -45,6 +45,7 @@ Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}() Base.length(g::TensorGrid) = sum(length, g.grids) Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...) +Base.size(g::TensorGrid, d) = size(g)[d] refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids)