Merge pull request #65 from kc-ml2/DEV/main

update vector modeling; change rectangle method;
kc-ml2 · Apr 11, 2024 · 21c445a · 21c445a
2 parents fc0943e + 35f9103
commit 21c445a
Show file tree

Hide file tree

Showing 4 changed files with 434 additions and 129 deletions.
diff --git a/QA/grad-vector-rectangle-rotate.py b/QA/grad-vector-rectangle-rotate.py
@@ -41,7 +41,7 @@ def forward(length1, length2, angle, c1):
     length1 = length1.type(torch.complex128)
     length2 = length2.type(torch.complex128)
 
-    obj_list = mee.rectangle_rotate(*c1, length1, length2, 5, 5, n_index, angle)
+    obj_list = mee.rectangle(*c1, length1, length2, 5, 5, n_index, angle)
 
     layer_info_list = [[layer_base, obj_list]]
 

diff --git a/QA/grad-vector.py b/QA/grad-vector.py
@@ -42,8 +42,8 @@ def forward(input_length1, input_length2, input_length3, input_length4, period,
     length3 = input_length3.type(torch.complex128)
     length4 = input_length4.type(torch.complex128)
 
-    obj1_list = ModelingTorch.rectangle(*c1, length1, length2, n_index)
-    obj2_list = ModelingTorch.rectangle(*c2, length3, length4, n_index + 2)
+    obj1_list = ModelingTorch.rectangle_no_approximation(*c1, length1, length2, n_index)
+    obj2_list = ModelingTorch.rectangle_no_approximation(*c2, length3, length4, n_index + 2)
 
     obj_list = obj1_list + obj2_list