TridiagonalSolve

публичный финальный класс TridiagonalSolve

Решает трехдиагональные системы уравнений.

Решает трехдиагональные системы уравнений. Поддерживает размеры партии и несколько правых сторон для каждой левой стороны. На ЦП решение вычисляется методом исключения Гаусса с частичным поворотом или без него, в зависимости от атрибута «partial_pivoting». На графическом процессоре используется библиотека cuSPARSE от Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Частичный поворот пока не поддерживается серверными модулями XLA.

Вложенные классы

сорт TridiagonalSolve.Options Дополнительные атрибуты для TridiagonalSolve

Константы

Нить OP_NAME Название этой операции, известное основному движку TensorFlow.

Публичные методы

Выход <Т>
какВывод ()
Возвращает символический дескриптор тензора.
static <T расширяет TType > TridiagonalSolve <T>
create ( Область видимости , Диагонали операнда <T>, Правая часть операнда <T>, Параметры... )
Фабричный метод для создания класса, обертывающего новую операцию TridiagonalSolve.
Выход <Т>
выход ()
Тензор формы `[..., M, K]`, содержащий решения
статический TridiagonalSolve.Options
частичное сведение (логическое частичное сведение)

Унаследованные методы

Константы

общедоступная статическая финальная строка OP_NAME

Название этой операции, известное основному движку TensorFlow.

Постоянное значение: «TridiagonalSolve».

Публичные методы

публичный вывод <T> asOutput ()

Возвращает символический дескриптор тензора.

Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.

public static TridiagonalSolve <T> create ( Область действия, диагонали операнда <T>, операнд <T> справа, параметры... параметры)

Фабричный метод для создания класса, обертывающего новую операцию TridiagonalSolve.

Параметры
объем текущий объем
диагонали Тензор формы `[..., 3, M]`, два внутренних измерения которого представляют трехдиагональные матрицы, три строки которых представляют собой супердиагонали, диагонали и субдиагонали по порядку. Последний элемент наддиагонали и первый элемент поддиагонали игнорируются.
правый риск Тензор формы `[..., M, K]`, представляющий K правых частей на каждую левую часть.
параметры содержит значения необязательных атрибутов
Возврат
  • новый экземпляр TridiagonalSolve

публичный вывод <T> вывод ()

Тензор формы `[..., M, K]`, содержащий решения

общедоступный статический TridiagonalSolve.Options частичное сведение (логическое значение частичного свертывания)

Параметры
частичный поворот Применять ли частичный поворот. Частичный поворот делает процедуру более стабильной, но более медленной.