Mercurial > repos > public > sbplib_julia

comparison test/LazyTensors/lazy_tensor_operations_test.jl @ 1837:200971c71657 refactor/lazy_tensors/elementwise_ops

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Refactor ElementwiseTensorOperation into TensorSum and use TensorNegation for handling subtraction
author Jonatan Werpers <jonatan@werpers.com>
date Thu, 09 Jan 2025 21:46:01 +0100
parents 368999a2e243
children 4bd998069053
comparison
equal deleted inserted replaced
1836:368999a2e243 1837:200971c71657
18 end 18 end
19 19
20 LazyTensors.apply(m::SizeDoublingMapping{T,R}, v, i::Vararg{Any,R}) where {T,R} = (:apply,v,i) 20 LazyTensors.apply(m::SizeDoublingMapping{T,R}, v, i::Vararg{Any,R}) where {T,R} = (:apply,v,i)
21 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size 21 LazyTensors.range_size(m::SizeDoublingMapping) = 2 .* m.domain_size
22 LazyTensors.domain_size(m::SizeDoublingMapping) = m.domain_size 22 LazyTensors.domain_size(m::SizeDoublingMapping) = m.domain_size
23
24 23
25 24
26 @testset "Mapping transpose" begin 25 @testset "Mapping transpose" begin
27 m = TransposableDummyMapping{Float64,2,3}() 26 m = TransposableDummyMapping{Float64,2,3}()
28 @test m' isa LazyTensor{Float64, 3,2} 27 @test m' isa LazyTensor{Float64, 3,2}
154 153
155 @test range_size(-Ã) == (2,) 154 @test range_size(-Ã) == (2,)
156 @test range_size(-B̃) == (3,) 155 @test range_size(-B̃) == (3,)
157 end 156 end
158 157
159 @testset "LazyTensor binary operations" begin 158 @testset "TensorSum" begin
160 A = ScalingTensor(2.0, (3,)) 159 A = ScalingTensor(2.0, (3,))
161 B = ScalingTensor(3.0, (3,)) 160 B = ScalingTensor(3.0, (3,))
162 161
163 v = [1.1,1.2,1.3] 162 v = [1.1,1.2,1.3]
164 for i ∈ eachindex(v) 163 for i ∈ eachindex(v)