simonfranken/mmp_wise2526_franksim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

main: fix

Browse Source
This commit is contained in:
franksim
2025-10-20 13:04:19 +00:00
parent d173712d9d
commit 61edc12522
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mmp/a1/tensors.py
View File

@@ -14,4 +14,4 @@ def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.
def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor): def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor):
return matrix.add(vector) return matrix + vector.unsqueeze(1)