TensorFlow维度变换函数语句

学习课程
1.reshape

a = tf.random.normal([4,28,28,3]) #4张照片
a.shape,a.ndim  #TensorShape([4,28,28,3]),4
tf.reshape(a,[4,784,3]).shape  #TensorShape([4,784,3])
tf.reshape(a,[4,-1,3]).shape  #TensorShape([4,784,3])
tf.reshape(a,[4,-1]).shape  #TensorShape([4,2352])
tf.reshae(tf.reshape(a,[4,-1])[4,1,784,3]).shape  #TensorShape([4,1,784,3])

Reshape could lead to potential bugs!

2.tf.transpose

a = tf.random.normal([4,28,56,3])
tf.transpose(a).shape   #TensorShape([3,56,28,4])
tf.transpose(a,perm=[0,1,3,2]).shape   #TensorShape([4,28,3,56])

3.Expand dim

a = tf.random.normal([4,35,8])
tf.expand_dims(a,axis=0).shape  #TensorShape([1,4,35,8])
tf.expand_dims(a,axis=1).shape  #TensorShape([4,1,35,8])
tf.expand_dims(a,axis=-1).shape  #TensorShape([4,35,8,1])
tf.expand_dims(a,axis=-4).shape  #TensorShape([1,4,35,8])

4.Squeeze dim

a = tf.random.normal([1,1,4,35,1,8])
tf.squeeze(a,axis=0).shape  #TensorShape([1,4,35,1,8])
tf.squeeze(a,axis=4).shape  #TensorShape([1,1,4,35,8])
tf.squeeze(a,axis=-2).shape  #TensorShape([1,1,4,35,8])

5.Broadcasting

x=tf.random.normal([4,32,32,3])
x+tf.random.normal([3]).shape  #TensorShape([4,32,32,3])
x+tf.random.normal([32,32,1]).shape  #TensorShape([4,32,32,3])
x+tf.random.normal([4,1,1,1]).shape  #TensorShape([4,32,32,3])

6.tf.broadcast_to

b=tf.broadcast_to(tf.random.normal([4,1,1,1]),[4,32,32,3]).shape    #TensorShape([4,32,32,3])

TensorFlow维度变换函数语句相关推荐

TensorFlow——维度变换与Broadcasting
TensorFlow 维度变换文章目录 TensorFlow 维度变换一.Reshape 二.tf.transpose 三.Squeeze和Expand_dims Broadcasting 前言 ...
tensorflow随笔-条件语句-tf.cond
tf.cond tf.cond( pred, true_fn=None, false_fn=None, strict=False, name=None, fn1=None, fn2=None ) 如果 ...
tensorflow随笔-条件语句-tf.case
tf.case tf.case( pred_fn_pairs, default=None, exclusive=False, strict=False, name='case' ) 创建case操作 ...
TensorFlow创建tensor语句
学习课程 1.np.array转tf: tf.convert_to_tensor(np.ones(2,1)) tf.convert_to_tensor([2,1]) tf.convert_to_ten ...
python tensorflow 智能家居_TensorFlow平台下的视频目标跟踪深度学习模型设计
4.3 深度学习模型设计实现实验开始阶段需要链接相关函数库, 程序中采用如下的语句: import TensorFlow as tf import cv - import TensorFlow as ...
Tensorflow
本文参考 1)戈云飞 https://blog.csdn.net/geyunfei_/article/details/78782804 2)量化大司马 https://www.jianshu. ...
TensorFlow 1.x 深度学习秘籍：1~5
原文:TensorFlow 1.x Deep Learning Cookbook 协议:CC BY-NC-SA 4.0 译者:飞龙本文来自[ApacheCN 深度学习译文集],采用译后编辑(MTP ...
如何看待「TensorFlow就是一颗定时炸弹」的说法？
链接:https://www.zhihu.com/question/397135189 点击上方"AI遇见机器学习",选择"星标"公众号重磅干货,第一时间送达 ...
关于tensorflow的碎片
1.突然间可视化tensorflow报错:ImportError: cannot import name 'monitoring' 解决: pip uninstall tensorflow_estim ...

TensorFlow维度变换函数语句

TensorFlow维度变换函数语句相关推荐

最新文章

热门文章