理解Tensorflow的shape

Tensorflow中的shape应该怎么理解,怎么记住呢?

    x = tf.placeholder(tf.float32, shape=[1,2,3] )

以上面这段代码为例为例。
首先,单看[1,2,3],这是1维的,但它作为shape时,代表要传入的数据必须是个3维的,这点首先要明白,自己理解一下。

Tensorflow和numpy一样,读shape时应该从外向内读。

先举个例子:
[[1,2,3] , [4,5,6]]

[[1,2,3],
[4,5,6]]
是一样的,都是2行3列(shape=[2,3])。应该怎么记呢?
这个矩阵,先拿掉最外层中括号,变成[1,2,3], [4,5,6],[1,2,3]和 [4,5,6]被逗号隔开成2块,理解为有2个元素,每个元素(如[1,2,3])拿掉中括号后,剩下1、2和3被逗号隔开,理解为有3个元素,所以是shape=[2,3]。

再换个例子,如果shape=[1,1,1],那它会接收什么样的数据?我们根据规则,第1个数字为“1”表示最外层的元素个数只有1个。
[a]
第二层的数字为“1”表示拿掉一次括号后,剩下的仍然只有1个元素.
[ [a] ]
相应的,第3个“1”表示再拿掉一次括号后还是只剩1个元素
[[ [a] ]]就是结果。

shape = [1,1,2]表示数据应该是这样的:[[[a,b]]]。

回到最开始,x应该输入的是[[[a,b,c],[d,e,f]]]这样格式的。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值