Решает трехдиагональные системы уравнений.
Решает трехдиагональные системы уравнений. Поддерживает размеры партии и несколько правых сторон для каждой левой стороны. На ЦП решение вычисляется методом исключения Гаусса с частичным поворотом или без него, в зависимости от атрибута «partial_pivoting». На графическом процессоре используется библиотека cuSPARSE от Nvidia: https://docs.nvidia.com/cuda/cusparse/index.html#gtsv Частичный поворот пока не поддерживается серверными модулями XLA.
Вложенные классы
| сорт | TridiagonalSolve.Options | Дополнительные атрибуты для TridiagonalSolve | |
Публичные методы
| Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
| статический <T> Трехдиагональное решение <T> | create ( Область видимости , Диагонали операнда <T>, Правая часть операнда <T>, Параметры... ) Фабричный метод для создания класса, обертывающего новую операцию TridiagonalSolve. |
| Выход <Т> | выход () Тензор формы `[..., M, K]`, содержащий решения |
| статический TridiagonalSolve.Options | частичное сведение (логическое частичное сведение) |
Унаследованные методы
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static TridiagonalSolve <T> create ( Область действия, диагонали операнда <T>, операнд <T> справа, параметры... параметры)
Фабричный метод для создания класса, обертывающего новую операцию TridiagonalSolve.
Параметры
| объем | текущий объем |
|---|---|
| диагонали | Тензор формы `[..., 3, M]`, два внутренних измерения которого представляют трехдиагональные матрицы, три строки которых представляют собой супердиагональ, диагонали и субдиагонали по порядку. Последний элемент наддиагонали и первый элемент поддиагонали игнорируются. |
| правый риск | Тензор формы `[..., M, K]`, представляющий K правых частей на каждую левую часть. |
| параметры | содержит значения необязательных атрибутов |
Возврат
- новый экземпляр TridiagonalSolve
общедоступный статический TridiagonalSolve.Options частичное сведение (логическое значение частичного свертывания)
Параметры
| частичный поворот | Применять ли частичный поворот. Частичный поворот делает процедуру более стабильной, но более медленной. |
|---|