基于pytorch的少样本故障诊断注意力机制