Tensorflow算子tf.keras.layers.Reshape&tf.reshape

1. tf.keras.layers.Reshape会变成一个layer, 而tf.reshape只是一个计算;用tf.reshape训练假设得到100层的模型,tf.keras.reshape训练五层会有105层模型;

2. tf.keras.layers.Reshape会消耗更多的算力和内存;

posted @ 2024-11-12 09:59  张幼安  阅读(15)  评论(0编辑  收藏  举报