tf.keras.backend.ndim
| Просмотреть исходный код на GitHub |
Возвращает количество осей в тензоре в виде целого числа.
tf.keras.backend.ndim(
x
)
| Аргументы | |
|---|---|
x | Тензор или переменная. |
| Возвращаемое значение | |
|---|---|
| Целое число (скаляр), количество осей. |
Примеры:
from keras import backend as K input = K.placeholder(shape=(2, 4, 5)) val = np.array([[1, 2], [3, 4]]) kvar = K.variable(value=val) K.ndim(input) 3 K.ndim(kvar) 2
© 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/ndim