Mercurial > repos > public > sbplib_julia
comparison src/Grids/tensor_grid.jl @ 1423:48e16efaac7a bugfix/grids/tensor_grid_length
Fix implementation of `length(::TensorGrid)`
author | Jonatan Werpers <jonatan@werpers.com> |
---|---|
date | Thu, 07 Sep 2023 16:31:29 +0200 |
parents | e82240df974d |
children | 29adc06e5b56 c0c5e0620b83 |
comparison
equal
deleted
inserted
replaced
1422:949fe47d65a4 | 1423:48e16efaac7a |
---|---|
42 _iterate_combine_coords(::Nothing) = nothing | 42 _iterate_combine_coords(::Nothing) = nothing |
43 _iterate_combine_coords((next,state)) = combine_coordinates(next...), state | 43 _iterate_combine_coords((next,state)) = combine_coordinates(next...), state |
44 | 44 |
45 Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}() | 45 Base.IteratorSize(::Type{<:TensorGrid{<:Any, D}}) where D = Base.HasShape{D}() |
46 Base.eltype(::Type{<:TensorGrid{T}}) where T = T | 46 Base.eltype(::Type{<:TensorGrid{T}}) where T = T |
47 Base.length(g::TensorGrid) = sum(length, g.grids) | 47 Base.length(g::TensorGrid) = prod(length, g.grids) |
48 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...) | 48 Base.size(g::TensorGrid) = LazyTensors.concatenate_tuples(size.(g.grids)...) |
49 | 49 |
50 | 50 |
51 refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids) | 51 refine(g::TensorGrid, r::Int) = mapreduce(g->refine(g,r), TensorGrid, g.grids) |
52 coarsen(g::TensorGrid, r::Int) = mapreduce(g->coarsen(g,r), TensorGrid, g.grids) | 52 coarsen(g::TensorGrid, r::Int) = mapreduce(g->coarsen(g,r), TensorGrid, g.grids) |