tf.keras.ops.vectorized_map
Параллельное применение function по оси 0 тензора (тензоров) elements.
tf.keras.ops.vectorized_map(
function, elements
)
Схематически, vectorized_map реализует следующее для одного входного тензора elements:
def vectorized_map(function, elements)
outputs = []
for e in elements:
outputs.append(function(e))
return stack(outputs)
Для итерируемого объекта из тензоров elements реализуется следующее:
def vectorized_map(function, elements)
batch_size = elements[0].shape[0]
outputs = []
for index in range(batch_size):
outputs.append(function([e[index] for e in elements]))
return np.stack(outputs)
В этом случае ожидается, что function принимает в качестве входных данных один список аргументов-тензоров.
© 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/vectorized_map
