『tensorflow笔记』tf.reshape的详细讲解
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
函数原型:目的是为了功能改变张量(tensor)的形状。
tf.reshape(
tensor,
shape,
name=None
)
tensor
形参传入一个tensor。shape
传入一个向量,代表新tensor的维度数和每个维度的长度。如果传入[3,4,5]
,就会返回一个内含各分量数值和原传入张量一模一样的3*4*5尺寸的张量。如果
shape
传入的向量某一个分量设置为-1,比如[-1,4,5]
,那么这个分量代表的维度尺寸会被自动计算出来。
- 用法一,一个尺寸为1*9的张量转化为3*3的张量:
#coding:utf-8
import tensorflow as tf
t=[1,2,3,4,5,6,7,8,9]
with tf.Session() as sess:
print (sess.run(tf.reshape(t,[3,3])))
输出结果:
(tf14) zhangkf@Ubuntu2:~$ python o.py
[[1 2 3]
[4 5 6]
[7 8 9]]
- 用法二,一个尺寸为3 * 2 * 3的张量,转换为第二个维度尺寸为9的张量,即n*9的张量:
#coding:utf-8
import tensorflow as tf
t=[[[1,2],[1,2],[1,2]],[[1,2],[1,2],[1,2]],[[1,2],[1,2],[1,2]]]
with tf.Session() as sess:
print (sess.run(tf.reshape(t,[-1,9])))
输出结果:显然,n被计算为2。
(tf14) zhangkf@Ubuntu2:~$ python o.py
[[1 2 1 2 1 2 1 2 1]
[2 1 2 1 2 1 2 1 2]]
- 用法三,仅含有单个元素的张量转化为标量:
t
为张量[7]
#coding:utf-8
import tensorflow as tf
t=[7]
with tf.Session() as sess:
print (sess.run(tf.reshape(t,[])))
输出结果:
(tf14) zhangkf@Ubuntu2:~$ python o.py
7
_image = tf.reshape(x, [-1,28, 28, 1])
# -1表示任意数量的样本数,大小为28x28深度为一的张量
# 可以忽略(其实是用深度为28的,28x1的张量,来表示28x28深度为1的张量)
GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2
PiperOrigin-RevId: 663726708
2 个月前
91dac11a
This test overrides disabled_backends, dropping the default
value in the process.
PiperOrigin-RevId: 663711155
2 个月前
更多推荐
已为社区贡献9条内容
所有评论(0)