Update basic code

2024-07-22 21:17:49 +08:00
parent 401bd229b0
commit 13f660826d
4 changed files with 202 additions and 6 deletions
@@ -46,7 +46,11 @@ function x = ST(h_1, lambda, Q_1)
    x = zeros(N, M);

    for i = 1:N
-        sign = h_1(i) ./ abs(h_1(i));
+        if h_1(i) == 0
+            sign = 1;
+        else
+            sign = h_1(i) ./ abs(h_1(i));
+        end
        diff = abs(h_1(i)) - lambda;
        x(i) = sign .* (diff ./ Q_1) .* SF(diff);
    end