from dataclasses import dataclass

from torch import Tensor

@dataclass
class AttnFeatureInfo:
    k: Tensor
    qk: Tensor
    vqk: Tensor