did.nystrom

Implements generating Nystorm points (on a grid).

Module Contents

Functions

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.

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