Mercurial > repos > public > sbplib_julia

diff src/LazyTensors/tensor_mapping.jl @ 943:fb060e98ac0a feature/tensormapping_application_promotion

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Remove more type assertions
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 10 Mar 2022 16:57:01 +0100
parents 76e5682d0e52
children 4a4ef4bf6cb9
line wrap: on
line diff
--- a/src/LazyTensors/tensor_mapping.jl	Thu Mar 10 11:13:34 2022 +0100
+++ b/src/LazyTensors/tensor_mapping.jl	Thu Mar 10 16:57:01 2022 +0100
@@ -24,7 +24,7 @@
 export TensorMapping
 
 """
-    apply(t::TensorMapping{T,R,D}, v::AbstractArray{T,D}, I::Vararg) where {R,D,T}
+    apply(t::TensorMapping{T,R,D}, v::AbstractArray{<:Any,D}, I::Vararg) where {R,D,T}
 
 Return the result of the mapping for a given index.
 """
@@ -32,7 +32,7 @@
 export apply
 
 """
-    apply_transpose(t::TensorMapping{T,R,D}, v::AbstractArray{T,R}, I::Vararg) where {R,D,T}
+    apply_transpose(t::TensorMapping{T,R,D}, v::AbstractArray{<:Any,R}, I::Vararg) where {R,D,T}
 
 Return the result of the transposed mapping for a given index.
 """