Static Shape Constraint Verification
CHECKING
Static Verification Status: Sound (0 Type Errors)
Active Contract: Tensor[B=4, S=128, D=512] -> MultiHeadAttention -> Tensor[B=4, S=128, D=512]
Active Contract: Tensor[B=4, S=128, D=512] -> MultiHeadAttention -> Tensor[B=4, S=128, D=512]
PEP 695 / PyreFly Type Stub
Static Assertion Contract
# Generated PyreFly Tensor Contract Stub
from typing import TypeVar
import torch
from torch import Tensor
B = TypeVar("B") # Batch
S = TypeVar("S") # Sequence
D = TypeVar("D") # Dimension
H = TypeVar("H") # Heads