tensorflow查看形状XXX.get_shape()、numpy查看形状XXX.shape tf.matmul(a, w)+b详细解释(CNN全连接相关的)

import tensorflow as tf

a = tf.constant((1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0), dtype=tf.float32, shape=[4, 2], name='a')
aa = tf.reshape(a, shape=[-1, 4])
w = tf.ones(shape=[4, 2], dtype=tf.float32, name=None)
b = tf.ones(shape=[2], dtype=tf.float32, name=None)
with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(a))
    '''
    [[1. 2.]
     [3. 4.]
     [5. 6.]
     [7. 8.]]
    '''
    print(sess.run(aa))
    '''
    [[1. 2. 3. 4.]
     [5. 6. 7. 8.]]
    '''
    print(sess.run(w))
    '''
    [[1. 1.]
     [1. 1.]
     [1. 1.]
     [1. 1.]]
    '''
    print(sess.run(b))  # [1. 1.]
    print(b.get_shape())  # (2,)  tensorflow类型数据获取数据形状
    step = sess.run(tf.matmul(aa, w)+b)
    print(step)
    '''
    [[11. 11.]
     [27. 27.]]
    '''
    print(step.shape)  # (2, 2) numpy类型数据获取数据形状

    c = tf.argmax(step, axis=1)
    print(sess.run(c))  # [0 0]
    


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值