Skip to content

关于 Embedded Gaussian问题 #45

@mymuli

Description

@mymuli

想问一下,在non_local_embedded_gaussian.py实现文件中,我并没有发现Embedded Gaussian的具体表达式,只是有矩阵乘法+softmax函数...如果是Embedded Gaussian的实现,具体是哪几行代码实现呢?

f = torch.matmul(theta_x, phi_x)

f_div_C = F.softmax(f, dim=-1)

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions