计算机视觉中自注意力构建块的PyTorch实现