3D and higher reduction

JuliaReinforcementLearning · Aug 8, 2023 · 8aab70b · 8aab70b
1 parent c78e3b5
commit 8aab70b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/CircularArrayBuffers.jl b/src/CircularArrayBuffers.jl
@@ -96,7 +96,7 @@ end
     end
 end
 
-_buffer_frame(cb::CircularArrayBuffer, I::CartesianIndex) = map(i -> _buffer_frame(cb, i), Tuple(I)...)
+_buffer_frame(cb::CircularArrayBuffer, I::CartesianIndex) = CartesianIndex(map(i->_buffer_frame(cb, i), Tuple(I)))
 _buffer_frame(cb::CircularArrayBuffer, I::AbstractArray{<:Integer}) = map(i -> _buffer_frame(cb, i), I)
 _buffer_frame(cb::CircularArrayBuffer, ::Colon) = _buffer_frame(cb, axes(cb)[end])