added divide function

Author: rittermarc

src/tensortrain.jl

@@ -155,6 +155,21 @@ function Base.:*(a, tt::TensorTrain{V, N})::TensorTrain{V, N} where {V, N}
     return multiply(a, tt)
 end
 
+function divide!(tt::TensorTrain{V, N}, a) where {V, N}
+    tt.sitetensors[end] .= tt.sitetensors[end] ./ a
+    nothing
+end
+
+function divide(tt::TensorTrain{V, N}, a) where {V, N}
+    tt2 = deepcopy(tt)
+    divide!(tt2, a)
+    return tt2
+end
+
+function Base.:/(tt::TensorTrain{V, N}, a) where {V, N}
+    return divide(tt, a)
+end
+
 """
 Fitting data with a TensorTrain object.
 This may be useful when the interpolated function is noisy.

test/test_tensortrain.jl

@@ -137,6 +137,9 @@ end
     tt1mul = 1.6 * tt1
     @test tt1mul.(indices) ≈ 1.6 .* tt1.(indices)
 
+    tt1div = tt1mul / 3.2
+    @test tt1div.(indices) ≈ tt1.(indices) ./ 2.0
+
     ttshort = TCI.TensorTrain{T,3}([randn(T, linkdims[n], localdims[n], linkdims[n+1]) for n in 1:L-1])
     @test_throws DimensionMismatch TCI.add(tt1, ttshort)