slot-attention:GoogleAI实现广告位关注

插槽注意 实施从纸“对象为中心的学习与插槽警示”中Pytorch。这是一段介绍此网络功能的。 更新：官方存储库已在发布 安装 $ pip install slot_attention 用法 import torch from slot_attention import SlotAttention slot_attn = SlotAttention ( num_slots = 5 , dim = 512 , iters = 3 # iterations of attention, defaults to 3 ) inputs = torch . randn ( 2 , 1024 , 512 ) slot_attn ( inputs ) # (2, 5, 512) 训练后，据报道该网络能够推广到稍有不同数量的插槽（群集）。您可以向前覆盖num_slots关键字使用的