nodes/mcDandy-more_math
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix node being able to be used

Browse Source
This commit is contained in:
mcDandy
2026-02-28 17:43:52 +01:00
parent c40f3e32b2
commit 0faaec395a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
more_math/Parser/UnifiedMathVisitor.py
+4 -4
View File
@@ -2539,10 +2539,10 @@ class UnifiedMathVisitor(MathExprVisitor):
def _bitwise_op(self, a, b, torch_op, scalar_op,ctx):
"""Binary bitwise operation handler supporting tensors, lists, and scalars."""
if self._is_tensor(a) and a.numel() == 1:
a = int(a.flatten()[0].item())
if self._is_tensor(b) and b.numel() == 1:
b = int(b.flatten()[0].item())
if self._is_tensor(a) and a.numel() == 1:
a = int(a.flatten()[0].item())
if self._is_tensor(b) and b.numel() == 1:
b = int(b.flatten()[0].item())
# Handle tensor-list combinations
if self._is_tensor(a) and self._is_list(b):