Mercurial > repos > public > sbplib_julia

diff src/Grids/tensor_grid.jl @ 1437:e3a80ef08d09

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Implement size(g,d) for EquidistantGrid and TensorGrid
author Jonatan Werpers <jonatan@werpers.com>
date Wed, 22 Nov 2023 17:34:35 +0100
parents e82240df974d
children 66ad232e45a1
line wrap: on
line diff
--- a/src/Grids/tensor_grid.jl	Fri Aug 25 08:49:07 2023 +0200
+++ b/src/Grids/tensor_grid.jl	Wed Nov 22 17:34:35 2023 +0100
@@ -46,6 +46,7 @@
 Base.eltype(::Type{<:TensorGrid{T}}) where T = T
 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)