tf.keras.ops.vectorize
Преобразует функцию в векторизованную функцию.
tf.keras.ops.vectorize(
pyfunc, *, excluded=None, signature=None
)
Пример:
def myfunc(a, b):
return a + b
vfunc = np.vectorize(myfunc)
y = vfunc([1, 2, 3, 4], 2) # Returns Tensor([3, 4, 5, 6])
| Аргументы | |
|---|---|
pyfunc | Вызываемый объект с одним аргументом типа тензор. |
excluded | Необязательный набор целых чисел, обозначающих позиционные аргументы, для которых функция не будет векторизована. Они будут переданы непосредственно в pyfunc без изменений. |
signature | Необязательная обобщённая сигнатура универсальной функции, например "(m,n),(n)->(m)" для векторизованного умножения матрицы на вектор. Если указана, pyfunc будет вызвана с массивами размерностей, заданными размерами соответствующих основных измерений, и должна возвращать такие массивы. По умолчанию предполагается, что pyfunc принимает на вход скалярные тензоры и возвращает их. |
| Возвращает | |
|---|---|
Новую функцию, которая применяет pyfunc к каждому элементу входных данных вдоль оси 0 (оси пакета). |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/keras/ops/vectorize
