Скопируйте тензор, устанавливающий все за пределами центральной полосы в каждой самой внутренней матрице в ноль.
Часть `band` вычисляется следующим образом: предположим, что `вход` имеет `k` измерений `[I, J, K, ..., M, N]`, тогда выходной сигнал представляет собой тензор той же формы, где
`band[i, j, k, ..., m, n] = in_band(m, n) * input[i, j, k, ..., m, n]`.
Индикаторная функция
`in_band(m, n) = (num_lower < 0 || (mn) <= num_lower)) && (num_upper < 0 || (nm) <= num_upper)`.
Например:
# if 'input' is [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[-2, -1, 0, 1]
[-3, -2, -1, 0]],
tf.matrix_band_part(input, 1, -1) ==> [[ 0, 1, 2, 3]
[-1, 0, 1, 2]
[ 0, -1, 0, 1]
[ 0, 0, -1, 0]],
tf.matrix_band_part(input, 2, 1) ==> [[ 0, 1, 0, 0]
[-1, 0, 1, 0]
[-2, -1, 0, 1]
[ 0, -2, -1, 0]]
tf.matrix_band_part(input, 0, -1) ==> Upper triangular part.
tf.matrix_band_part(input, -1, 0) ==> Lower triangular part.
tf.matrix_band_part(input, 0, 0) ==> Diagonal.
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
Выход <Т> | группа () Тензор ранга `k` той же формы, что и входные данные. |
static <T расширяет TType , U расширяет TNumber > BandPart <T> |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
публичный выходной диапазон <T> ()
Тензор ранга `k` той же формы, что и входные данные. Извлеченный полосатый тензор.
public static BandPart <T> create (область действия , ввод операнда <T>, операнд <U> numLower, операнд <U> numUpper)
Фабричный метод для создания класса, обертывающего новую операцию BandPart.
Параметры
объем | текущий объем |
---|---|
вход | Тензор ранга `k`. |
numLower | 0-D тензор. Количество сохраняемых субдиагоналей. Если отрицательно, сохраните весь нижний треугольник. |
numUpper | 0-D тензор. Количество сохраняемых супердиагоналей. Если отрицательно, сохраните весь верхний треугольник. |
Возврат
- новый экземпляр BandPart