from dataclasses import dataclass
from torch import Tensor
@dataclass
class AttnFeatureInfo:
qk: Tensor
vqk: Tensor