题目
'''
Description: transformer[Attention机制]
Autor: 365JHWZGo
Date: 2022-01-04 16:16:41
LastEditors: 365JHWZGo
LastEditTime: 2022-01-04 21:56:26
'''
引用库
import torch.nn as nn
from torch.nn.functional import cross_entropy, softmax, relu
import numpy as np
import torch
from torch