Миграция с TF1 на TF2 - PullRequest
Новая
       26

Миграция с TF1 на TF2

0 голосов
/ 03 февраля 2020

Я хотел бы перейти с tf1 на TF2, поэтому я вижу этот код ниже (после конвертации), но я не могу понять, почему tf.ones? зная, что эта функция устанавливает все элементы в один, 

model = tf.keras.Sequential([
    tf.keras.layers.Conv2D(32, 3, activation='relu',
                           kernel_regularizer=tf.keras.regularizers.l2(0.04),
                           input_shape=(28, 28, 1)),
    tf.keras.layers.MaxPooling2D(),
    tf.keras.layers.Flatten(),
    tf.keras.layers.Dropout(0.1),
    tf.keras.layers.Dense(64, activation='relu'),
    tf.keras.layers.BatchNormalization(),
    tf.keras.layers.Dense(10, activation='softmax')
])

train_data = tf.ones(shape=(1, 28, 28, 1))
test_data = tf.ones(shape=(1, 28, 28, 1))

train_out = model(train_data, training=True)

test_out = model(test_data, training=False)

вот учебник: https://www.tensorflow.org/guide/migrate

...