Mercurial > repos > public > sbplib_julia

diff src/LazyTensors/lazy_tensor_operations.jl @ 977:043d13ef8898 feature/tensormapping_application_promotion

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Fix type calculation of LazyTensorMappingApplication element type
author Jonatan Werpers <jonatan@werpers.com>
date Tue, 15 Mar 2022 21:03:17 +0100
parents 089a1411dfc0
children df562695b1b5 bc384aaade30 bbbc31953367
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl	Tue Mar 15 07:40:13 2022 +0100
+++ b/src/LazyTensors/lazy_tensor_operations.jl	Tue Mar 15 21:03:17 2022 +0100
@@ -12,7 +12,8 @@
     o::AA
 
     function LazyTensorMappingApplication(t::TensorMapping{<:Any,R,D}, o::AbstractArray{<:Any,D}) where {R,D}
-        T = promote_type(eltype(t), eltype(o))
+        I = ntuple(i->1, range_dim(t))
+        T = typeof(apply(t,o,I...))
         return new{T,R,D,typeof(t), typeof(o)}(t,o)
     end
 end