摘要: class AxialAttention(nn.Module): def __init__( self, dim, heads, row_attn = True, col_attn = True, accept_edges = False, global_query_attn = False, ** 阅读全文
posted @ 2024-10-24 19:33 GraphL 阅读(16) 评论(0) 推荐(0)