We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 8411f53 + 5b5b849 commit 2581110Copy full SHA for 2581110
tester/paddle_to_torch/rules.py
@@ -1527,7 +1527,7 @@ def apply(self, paddle_api: str) -> ConvertResult:
1527
core = "result = x.diag(diagonal=offset)"
1528
post = """
1529
if x.ndim == 1 and padding_value != 0:
1530
- padding_value = torch.tensor(padding_value, dtype=x.dtype)
+ padding_value = torch.tensor(padding_value, dtype=torch.float32)
1531
diag_mask = torch.diag(torch.ones_like(x), diagonal=offset)
1532
result = torch.where(diag_mask.bool(), result, padding_value)
1533
"""
0 commit comments