Mercurial > repos > public > sbplib_julia
diff test/LazyTensors/lazy_tensor_operations_test.jl @ 1836:368999a2e243 refactor/lazy_tensors/elementwise_ops
Add TensorNegation
| author | Jonatan Werpers <jonatan@werpers.com> |
|---|---|
| date | Thu, 09 Jan 2025 15:32:47 +0100 |
| parents | 471a948cd2b2 |
| children | 200971c71657 |
line wrap: on
line diff
--- a/test/LazyTensors/lazy_tensor_operations_test.jl Thu Jan 09 12:40:49 2025 +0100 +++ b/test/LazyTensors/lazy_tensor_operations_test.jl Thu Jan 09 15:32:47 2025 +0100 @@ -128,6 +128,33 @@ end end +@testset "TensorNegation" begin + A = rand(2,3) + B = rand(3,4) + + Ã = DenseTensor(A, (1,), (2,)) + B̃ = DenseTensor(B, (1,), (2,)) + + @test -Ã isa TensorNegation + + v = rand(3) + @test (-Ã)*v == -(Ã*v) + + v = rand(4) + @test (-B̃)*v == -(B̃*v) + + v = rand(2) + @test (-Ã)'*v == -(Ã'*v) + + v = rand(3) + @test (-B̃)'*v == -(B̃'*v) + + @test domain_size(-Ã) == (3,) + @test domain_size(-B̃) == (4,) + + @test range_size(-Ã) == (2,) + @test range_size(-B̃) == (3,) +end @testset "LazyTensor binary operations" begin A = ScalingTensor(2.0, (3,))