import torch


def D_power_bias_X(graph, X, power, coeff, bias):
    """Y = (coeff*D + bias*I)^{power} X"""
    degs = graph.ndata["deg"]
    degs = coeff * degs + bias
    norm = torch.pow(degs, power)
    Y = X * norm.view(X.size(0), 1)
    return Y