Database, Tech.
labels=None): hit_item_emb = self.item_emb(hist_item) # [B, seqlen, embed_dim] user_cap, cap_weights, cap_mask = self.capsual_layer(hit_item_emb, seqlen) if not self.training: return user_cap, cap_weights target_emb = self.item_emb(labels) user_emb, W = self.label_aware_attention(user_cap,