did.nystrom¶
Implements generating Nystorm points (on a grid).
Module Contents¶
Functions¶
|
Generate Nystrom points grid in several dimensions. |
-
did.nystrom.compute_nystrom_grid(n_per_dim: int, dims: int, min_value: float, max_value: float, dtype) → torch.Tensor¶ Generate Nystrom points grid in several dimensions.
- Parameters
n_per_dim (int) – Number of Nyström points per dimension.
dims (int) – Number of dimensions.
min_value (float) – Lower bound for grid.
max_value (float) – Upper bound for grid (with np.linspace conventions).
dtype (object that is understood by dtype=dtype in torch) – Usually torch.float32 or torch.float64.
- Returns
- Return type
torch.Tensor, Nystrom points grid