SobolEngine
-
class torch.quasirandom.SobolEngine(dimension, scramble=False, seed=None)[source] -
Двигатель
torch.quasirandom.SobolEngineпредназначен для генерации (перетасованных) последовательностей Соболя. Последовательности Соболя являются примером квазислучайных последовательностей с малой дисперсией.Эта реализация двигателя для последовательностей Соболя способна генерировать последовательности до максимальной размерности 21201. Она использует числа направлений из https://web.maths.unsw.edu.au/~fkuo/sobol/, полученные с использованием критерия поиска D(6) до размерности 21201. Это рекомендуемый выбор авторами.
Литература
- Art B. Owen. Перетасовка точек Соболя и Нидеррейтера-Сина. Журнал сложности, 14(4):466-489, декабрь 1998.
- И. М. Соболь. Распределение точек в кубе и точное вычисление интегралов. Ж. Вычисл. Матем. и Матем. Физики, 7:784-802, 1967.
- Параметры:
-
- dimension (Int) – Размерность генерируемой последовательности
-
scramble (bool, optional) – Установка этого значения в
Trueприведет к созданию перетасованных последовательностей Соболя. Перетасовка может улучшить последовательности Соболя. По умолчанию:False. -
seed (Int, optional) – Это семя для перетасовки. Если указано, то оно используется в качестве семени генератора случайных чисел. В противном случае используется случайное семя. По умолчанию:
None
Примеры:
>>> soboleng = torch.quasirandom.SobolEngine(dimension=5) >>> soboleng.draw(3) tensor([[0.0000, 0.0000, 0.0000, 0.0000, 0.0000], [0.5000, 0.5000, 0.5000, 0.5000, 0.5000], [0.7500, 0.2500, 0.2500, 0.2500, 0.7500]])-
draw(n=1, out=None, dtype=torch.float32)[source] -
Функция для генерации последовательности из
nточек из последовательности Соболя. Обратите внимание, что образцы зависят от предыдущих образцов. Размер результата равен .- Параметры:
-
- n (Int, optional) – Длина последовательности точек для генерации. По умолчанию: 1
- out (Tensor, optional) – Выходной тензор
-
dtype (
torch.dtype, optional) – Желаемый тип данных возвращаемого тензора. По умолчанию:torch.float32
- Тип возвращаемого значения:
-
draw_base2(m, out=None, dtype=torch.float32)[source] -
Функция для генерации последовательности из
2**mточек из последовательности Соболя. Обратите внимание, что образцы зависят от предыдущих образцов. Размер результата равен .- Параметры:
-
- m (Int) – (По основанию 2) показатель степени числа генерируемых точек.
- out (Tensor, optional) – Выходной тензор
-
dtype (
torch.dtype, optional) – Желаемый тип данных возвращаемого тензора. По умолчанию:torch.float32
- Тип возвращаемого значения:
-
fast_forward(n)[source] -
Функция для быстрого продвижения состояния
SobolEngineнаnшагов. Это эквивалентно генерацииnобразцов без использования самих образцов.- Параметры:
-
n (Int) – Количество шагов для быстрого продвижения.
-
reset()[source] -
Функция для сброса состояния
SobolEngineдо исходного состояния.
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.quasirandom.SobolEngine.html