mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
6f15a609d3
See #145838 Pull Request resolved: https://github.com/pytorch/pytorch/pull/146426 Approved by: https://github.com/Skylion007
39 lines
920 B
Python
39 lines
920 B
Python
# flake8: noqa
|
|
from typing import Any
|
|
from typing_extensions import assert_type
|
|
|
|
from torch import randn, Tensor
|
|
|
|
|
|
# See ../pass/arithmetic_ops.py for more information
|
|
|
|
TENSOR, INT, FLOAT = randn(3), 2, 1.5
|
|
|
|
assert_type(
|
|
INT & TENSOR, # E: Unsupported operand types for & ("int" and "Tensor") [operator]
|
|
Any,
|
|
)
|
|
assert_type(
|
|
INT | TENSOR, # E: Unsupported operand types for | ("int" and "Tensor") [operator]
|
|
Any,
|
|
)
|
|
assert_type(
|
|
INT ^ TENSOR, # E: Unsupported operand types for ^ ("int" and "Tensor") [operator]
|
|
Any,
|
|
)
|
|
|
|
assert_type(
|
|
FLOAT # E: Unsupported operand types for & ("float" and "Tensor") [operator]
|
|
& TENSOR,
|
|
Tensor,
|
|
)
|
|
assert_type(
|
|
FLOAT # E: Unsupported operand types for | ("float" and "Tensor") [operator]
|
|
| TENSOR,
|
|
Tensor,
|
|
)
|
|
assert_type(
|
|
FLOAT # E: Unsupported operand types for ^ ("float" and "Tensor") [operator]
|
|
^ TENSOR,
|
|
Tensor,
|
|
)
|