mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
[ONNX] Update types in VerificationInfo (#149377)
torch.types.Number was rendered as is in the documentation and can be confusing. We write the original types instead to reduce confusion for users. Pull Request resolved: https://github.com/pytorch/pytorch/pull/149377 Approved by: https://github.com/titaiwangms
This commit is contained in:
parent
405025778d
commit
fdacf3c920
|
|
@ -61,8 +61,8 @@ class VerificationInfo:
|
|||
def from_tensors(
|
||||
cls,
|
||||
name: str,
|
||||
expected: torch.Tensor | torch.types.Number,
|
||||
actual: torch.Tensor | torch.types.Number,
|
||||
expected: torch.Tensor | float | int | bool,
|
||||
actual: torch.Tensor | float | int | bool,
|
||||
) -> VerificationInfo:
|
||||
"""Create a VerificationInfo object from two tensors.
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user