tf.keras.backend.batch_flatten
| Просмотреть исходный код на GitHub |
Преобразует тензор nD в тензор 2D с одинаковым нулевым измерением.
tf.keras.backend.batch_flatten(
x
)
Другими словами, он сжимает каждый образец данных из пакета.
| Аргументы | |
|---|---|
x | Тензор или переменная. |
| Возвращает | |
|---|---|
| Тензор. |
Примеры:
Сжатие 3D тензора в 2D, сворачивая последнее измерение.
from tensorflow.keras import backend as K x_batch = K.ones(shape=(2, 3, 4, 5)) x_batch_flatten = K.batch_flatten(x_batch) K.int_shape(x_batch_flatten) (2, 60)
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/keras/backend/batch_flatten