tensorflow杂记

象象家的小蜜桃

已于 2022-12-16 20:39:34 修改

阅读量151

点赞数

分类专栏： tensorflow2使用文章标签： tensorflow 人工智能 python

于 2022-11-12 21:34:23 首次发布

本文链接：https://blog.csdn.net/qq_19069557/article/details/127825694

版权

tensorflow2使用专栏收录该内容

26 篇文章 10 订阅 ¥19.90 ¥99.00

订阅专栏

超级会员免费看

本文记录了在使用TensorFlow时的一些实用技巧，包括在创建层时为层命名、明确指定位置参数、Tensor与NumPy数组之间的转换，以及查看TensorFlow版本的方法。此外，还介绍了如何给数据添加新的轴和机器学习的基础术语。

摘要由CSDN通过智能技术生成

写层的时候记得取名字

写层的时候取名字是个好习惯，方面之后按照名字查找自己需要的层参数

取名字很方便，一般建立层的时候用到都是tf.keras.layers.Somelayer(...,name='layername')

Somelayer是指层的类型，比如Dense, Droupout, Conv2D,MaxPool2D,RNN等

调用时，使用name='layername'设置层的名称为layername

举个栗子

import tensorflow as tf
model = tf.keras.Sequential(
    [
        tf.keras.layers.Flatten(input_shape=(28,28),name='flatten_1'),
        tf.keras.layers.Dense(units=128, activation='relu',name='dense_1'),
        tf.keras.layers.Dropout(0.2,name='dropout_1'),
        tf.keras.layers.Dense(units=10,name='dense_2')
    ]
)

这样设置之后，想查看某个层的参数，可以使用如下方式