Source code for mlreco.models.layers.gnn.message_passing.dgtransformer

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F



[docs]class DeepGraphTransformer(nn.Module):


[docs]    def __init__(self, cfg):
        super(DeepGraphTransformer, self).__init__()