def METHOD_NAME(
    input_dims: int,
    n_harmonic_functions: int,
    append_input: bool,
) -> int:
    """
    Utility to help predict the shape of the output of `forward`.
    Args:
        input_dims: length of the last dimension of the input tensor
        n_harmonic_functions: number of embedding frequencies
        append_input: whether or not to concat the original
            input to the harmonic embedding
    Returns:
        int: the length of the last dimension of the output tensor
    """
    return input_dims * (2 * n_harmonic_functions + int(append_input))