Операция, поддерживающая базовую операцию einsum с 2 входами и 1 выходом.
Эта операция имеет лучшую производительность TPU, поскольку в ней нет явных операций изменения формы и транспонирования, как в tf.einsum.
Константы
Нить | OP_NAME | Название этой операции, известное основному движку TensorFlow. |
Публичные методы
Выход <Т> | какВывод () Возвращает символический дескриптор тензора. |
статический <T расширяет TType > Einsum <T> | |
Выход <Т> | продукт () |
Унаследованные методы
Константы
общедоступная статическая финальная строка OP_NAME
Название этой операции, известное основному движку TensorFlow.
Публичные методы
публичный вывод <T> asOutput ()
Возвращает символический дескриптор тензора.
Входные данные для операций TensorFlow являются выходными данными другой операции TensorFlow. Этот метод используется для получения символического дескриптора, который представляет собой вычисление входных данных.
public static Einsum <T> create (область действия , операнд <T> a, операнд <T> b, строковое уравнение)
Фабричный метод для создания класса, обертывающего новую операцию Einsum.
Параметры
объем | текущий объем |
---|
Возврат
- новый экземпляр Einsum