Mercurial > repos > public > sbplib_julia

diff src/Grids/tensor_grid.jl @ 1349:42ecd4b3e215 refactor/grids

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Add support for cartesian indices on TensorGrid
author Jonatan Werpers <jonatan@werpers.com>
date Fri, 19 May 2023 14:53:49 +0200
parents 08f06bfacd5c
children 4d628c83987e c0208286234e
line wrap: on
line diff
--- a/src/Grids/tensor_grid.jl	Fri May 19 14:31:16 2023 +0200
+++ b/src/Grids/tensor_grid.jl	Fri May 19 14:53:49 2023 +0200
@@ -26,6 +26,8 @@
     return vcat(ps...)
 end
 
+Base.getindex(g::TensorGrid, I::CartesianIndex) = g[Tuple(I)...]
+
 function Base.eachindex(g::TensorGrid)
     szs = LazyTensors.concatenate_tuples(size.(g.grids)...)
     return CartesianIndices(szs)