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.
1 parent b539e35 commit 5b5b849Copy full SHA for 5b5b849
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