(d, d_min=0.0, d_max=20.0, dim=16)
| 12 | |
| 13 | |
| 14 | def rbf(d, d_min=0.0, d_max=20.0, dim=16): |
| 15 | d_mu = torch.linspace(d_min, d_max, dim, device=d.device) |
| 16 | d_mu = d_mu.view([1, -1]) |
| 17 | d_sigma = (d_max - d_min) / dim |
| 18 | d_expand = torch.unsqueeze(d, -1) |
| 19 | |
| 20 | rbf = torch.exp(-((d_expand - d_mu) / d_sigma) ** 2) |
| 21 | return rbf |
| 22 | |
| 23 | |
| 24 | @R.register("models.SurfGVP") |
no outgoing calls
no test coverage detected