import torch


def compute_scores(users_embeds: torch.Tensor, items_embeds: torch.Tensor, items_bias: torch.Tensor) -> torch.Tensor:
    """
    Args:
        users_embeds(torch.Tensor): shape (batch_size, items_total, embed_dim)
        items_embeds(torch.Tensor): shape (items_total, embed_dim)
        items_bias(torch.Tensor): shape (items_total)

    Returns:
        scores(torch.Tensor): shape (batch_size, items_total)
    """
    scores = (users_embeds * items_embeds).sum(-1) + items_bias
    return scores