TensorFlow framework ML của Google. Keras API deep learning Python trên TensorFlow. Các lớp cần biết input shape để tạo trọng số.
Sử dụng Google Colab. Code minh họa: tạo Dense layer, in trọng số (ban đầu rỗng), gọi layer trên dữ liệu test, in trọng số sau khi tạo:
Ví dụ
print("Tạo ba lớp Dense")
layer = layers.Dense(3)
print("Trọng số liên kết với lớp")
print(layer.weights)
print("Gọi lớp trên dữ liệu test")
x = tf.ones((2, 3))
y = layer(x)
print("Bây giờ trọng số:")
print(layer.weights)
Nguồn: TensorFlow Guide
Đầu ra
Tạo ba lớp Dense Trọng số liên kết với lớp: [] Gọi lớp trên dữ liệu test Bây giờ trọng số: [, ]
Giải thích
- Mọi lớp Keras cần biết input shape để tạo trọng số tối ưu.
- Khi mới tạo, layer chưa có trọng số.
- Trọng số được tạo khi layer được gọi lần đầu trên input (vì phụ thuộc shape input).
- Các lớp được tạo tuần tự, gọi trên test data, in trọng số mới.