def METHOD_NAME(
    self,
    attributions,
    tokens,
    pred_prob,
    pred_class,
    true_class,
    attr_class,
    delta,
    vis_data_records,
):
    attributions = attributions.sum(dim=2).squeeze(0)
    attributions = attributions / torch.norm(attributions)
    attributions = attributions.cpu().detach().numpy()
    # storing couple samples in an array for visualization purposes
    vis_data_records.append(
        visualization.VisualizationDataRecord(
            attributions,
            pred_prob,
            pred_class,
            true_class,
            attr_class,
            attributions.sum(),
            tokens,
            delta,
        )
    )