变压器中的变压器：Pytorch中的Transformer在Transformer中的实现，像素级别注意与补丁级别注意配对以进行...

变压器中的变压器 Pytorch中实现，像素级别注意与补丁级别注意配对以进行图像分类。 安装 $ pip install transformer-in-transformer 用法 import torch from transformer_in_transformer import TNT tnt = TNT ( image_size = 256 , # size of image patch_dim = 512 , # dimension of patch token pixel_dim = 24 , # dimension of pixel token patch_size = 16 , # patch size pixel_size = 4 , # pixel size