aepsych.utils¶
aepsych.utils module¶
- aepsych.utils.dim_grid(lower, upper, dim, gridsize=30)[source]¶
Create a grid Create a grid based on either model dimensions, or pass in lower, upper, and dim separately. :param model: Input Model object that defines:
lower (‘int’) - lower bound
upper (‘int’) - upper bound
dim (‘int) - dimension
- Parameters
bound (- upper ('int') - upper) –
bound –
dimension (- dim ('int) -) –
grid (- gridsize ('int') - size for) –
lower (Tensor) –
upper (Tensor) –
dim (int) –
gridsize (int) –
- Returns
grid – Tensor
- Return type
torch.FloatTensor