diff src/LazyTensors/lazy_tensor_operations.jl @ 958:8b0ff2fddc32 feature/variable_derivatives

Merge default
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 14 Mar 2022 09:08:23 +0100
parents b41180efb6c2 6a2e30fa3bd8
children e9752c1e92f8
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl	Thu Mar 10 08:18:29 2022 +0100
+++ b/src/LazyTensors/lazy_tensor_operations.jl	Mon Mar 14 09:08:23 2022 +0100
@@ -15,6 +15,7 @@
 export LazyTensorMappingApplication
 
 Base.getindex(ta::LazyTensorMappingApplication{T,R}, I::Vararg{Any,R}) where {T,R} = apply(ta.t, ta.o, I...)
+Base.getindex(ta::LazyTensorMappingApplication{T,1}, I::CartesianIndex{1}) where {T} = apply(ta.t, ta.o, I.I...) # Would otherwise be caught in the previous method.
 Base.size(ta::LazyTensorMappingApplication) = range_size(ta.t)
 # TODO: What else is needed to implement the AbstractArray interface?