Spec-Zone.ru › PyTorch 1

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,dimension)(n, dimension).

Параметры:
  • n (Int, optional) – Длина последовательности точек для генерации. По умолчанию: 1
  • out (Tensor, optional) – Выходной тензор
  • dtype (torch.dtype, optional) – Желаемый тип данных возвращаемого тензора. По умолчанию: torch.float32
Тип возвращаемого значения:

Tensor

draw_base2(m, out=None, dtype=torch.float32) [source]

Функция для генерации последовательности из 2**m точек из последовательности Соболя. Обратите внимание, что образцы зависят от предыдущих образцов. Размер результата равен (2∗∗m,dimension)(2**m, dimension).

Параметры:
  • m (Int) – (По основанию 2) показатель степени числа генерируемых точек.
  • out (Tensor, optional) – Выходной тензор
  • dtype (torch.dtype, optional) – Желаемый тип данных возвращаемого тензора. По умолчанию: torch.float32
Тип возвращаемого значения:

Tensor

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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API