import torch


def angle_dist(vec_a: torch.Tensor, vec_b: torch.Tensor) -> torch.Tensor:
    """
    Combines vectors a and by concatenating [a * b, |a-b|]
    :param vec_a: A vector
    :param vec_b: Another vector with the same shape as vec_a
    :return: The combined vector
    """
    angle_vec = vec_a * vec_b
    dist_vec = torch.abs(vec_a - vec_b)
    return torch.cat((angle_vec, dist_vec), -1)