Mercurial > repos > public > sbplib_julia

diff src/LazyTensors/lazy_tensor_operations.jl @ 474:3ad327378b2d feature/outer_product

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Simplify implementation of \otimes
author Jonatan Werpers <jonatan@werpers.com>
date Mon, 26 Oct 2020 23:01:47 +0100
parents 0c3decc04649
children 2c0e76d5832d
line wrap: on
line diff
--- a/src/LazyTensors/lazy_tensor_operations.jl	Sun Oct 25 14:01:27 2020 +0100
+++ b/src/LazyTensors/lazy_tensor_operations.jl	Mon Oct 26 23:01:47 2020 +0100
@@ -326,7 +326,7 @@
 
 LazyOuterProduct(tms::Vararg{TensorMapping}) = foldl(LazyOuterProduct, tms)
 
-⊗(tms::Vararg{TensorMapping}) = LazyOuterProduct(tms...)
+⊗(a::TensorMapping, b::TensorMapping) = LazyOuterProduct(a,b)
 export ⊗
 
 # TODO: Can we implement compositions and kroneckers of LazyIdentities to just return new LazyIdentities?