
TensorFlow笔记
文章平均质量分 60
xLyons
小小研究生一枚,研究方向为推荐系统...
展开
-
TensorFlow2.x-教程链接
记录TensorFlow一些使用经验原创 2022-02-16 00:14:04 · 624 阅读 · 0 评论 -
tf.squeeze()和Flatten
如果某一维度为batch_size,代码会显示该维度为None;此时使用tf.squeeze()如果不指定axis,会出现压缩异常的情况,压缩后shape为unknown。理论上,会压缩所有维度为1的维度,但是这里压缩后却为unknown。如果指定了axis,则没有问题:由于这里使用Flatten也可以达到相同效果,所以最后代码为:Flatten()(tf.concat(embedding_query, axis=-1)但是Flatten是将所有的维度拉成一维,这个也要注意;...原创 2021-09-15 16:28:21 · 503 阅读 · 0 评论 -
tf.concat和tf.keras.layers.Concatenate()
二者的作用类似,不过使用起来也有区别:使用方式不同:tf.concat([obj1, obj2], axis=-1)tf.keras.layers.Concatenate(axis=-1)([obj1, obj2])concat拼接的对象类型要相同tf.float32和tf.float64似乎都无法直接拼接,反正我使用时会报错;...原创 2021-09-15 16:12:44 · 1489 阅读 · 0 评论 -
tf.keras.Layers中常见层的API接口
TensorFlow的官网API:https://tensorflow.google.cn/api_docs/python/tfkeras的中文网站API:https://keras-zh.readthedocs.io/这里只是记录一下常用的一些APItf.keras.LayersInput层tf.keras.Input( shape=None, batch_size=None, name=None, dtype=None, sparse=False, tensor=None, r原创 2021-02-05 22:54:30 · 1153 阅读 · 1 评论