import torch


def propagation(features, graph, k=1):
    """
    Return the features propagated k times on the graph.
    
    Params:
        features -- tensor of size (batch, num_features).
        graph -- adjacency matrix of size (num_features, num_features).
        k -- number of times the features are propagated, integer.
    """
    for _ in range(k):
        features = torch.matmul(features, graph)
    return features