@@ -309,16 +309,16 @@ end
309
309
struct FwdMap{N, E, T<: AbstractTangentBundle{N} }
310
310
f:: T
311
311
end
312
- FwdMap {E} (f:: T ) where {N, E, T<: AbstractTangentBundle{N} } = FwdMap {N,E,T} (f)
312
+ FwdMap {N, E} (f:: T ) where {N, E, T<: AbstractTangentBundle{N} } = FwdMap {N,E,T} (f)
313
313
(f:: FwdMap{N,E} )(args:: AbstractTangentBundle{N} ...) where {N,E} = ∂☆ {N,E} ()(f. f, args... )
314
314
315
315
function (:: ∂☆{N,E})(:: AbstractZeroBundle{N, typeof(map)} , f:: ATB{N} , tup:: TaylorBundle{N, <:Tuple} ) where {N,E}
316
- ∂vararg {N} ()(map (FwdMap {E} (f), destructure (tup))... )
316
+ ∂vararg {N} ()(map (FwdMap {N, E} (f), destructure (tup))... )
317
317
end
318
318
319
319
function (:: ∂☆{N,E})(:: AbstractZeroBundle{N, typeof(map)} , f:: ATB{N} , args:: ATB{N, <:AbstractArray} ...) where {N,E}
320
320
# TODO : This could do an inplace map! to avoid the extra rebundling
321
- rebundle (map (FwdMap {E} (f), map (unbundle, args)... ))
321
+ rebundle (map (FwdMap {N, E} (f), map (unbundle, args)... ))
322
322
end
323
323
324
324
function (:: ∂☆{N,E})(:: AbstractZeroBundle{N, typeof(map)} , f:: ATB{N} , args:: ATB{N} ...) where {N, E}
0 commit comments