tf.keras.Function

Класс, инкапсулирующий вычислительный граф операций Keras.

Наследуется от: Operation

tf.keras.Function(
    inputs, outputs, name=None
)

Вы можете использовать Function для захвата вычислительного графа, связывающего некоторые входные тензоры с некоторыми выходными тензорами, и повторного применения того же вычисления к новым входным данным.

Function похож на функциональную модель, с тем отличием, что он бессостоятельный (он не отслеживает переменные состояния) и не реализует API Layer.

Пример:

input_1 = keras.KerasTensor(shape=(None, 2, 3))
input_2 = keras.KerasTensor(shape=(None, 2, 3))
x = input_1 + input_2
output = keras.ops.sigmoid(x)
fn = keras.Function(inputs=[input_1, input_2], outputs=output)

input_1_val = np.random.random((4, 2, 3))
input_2_val = np.random.random((4, 2, 3))
output_val = fn([input_1_val, input_2_val])

Методы

call

Просмотреть исходный код

call(
    inputs
)

Вычисляет выходные тензоры для новых входных данных.

compute_output_spec

Просмотреть исходный код

compute_output_spec(
    inputs
)

from_config

Просмотреть исходный код

@classmethod
from_config(
    config
)

Создает слой из его конфигурации.

Этот метод является обратным к get_config, способным восстановить тот же слой из словаря конфигурации. Он не обрабатывает соединение слоев (обрабатывается Network), ни веса (обрабатываются set_weights).

get_config

Просмотреть исходный код

get_config()

Возвращает конфигурацию объекта.

Конфигурация объекта — это словарь Python (сериализуемый), содержащий информацию, необходимую для его повторного создания.

quantized_call

Просмотреть исходный код

quantized_call(
    *args, **kwargs
)

symbolic_call

Просмотреть исходный код

symbolic_call(
    *args, **kwargs
)

__call__

Просмотреть исходный код

__call__(
    *args, **kwargs
)

Вызов self как функции.