tensorflow随笔-矩阵乘法

版权声明：本博客所有文章版权归博主刘兴所有，转载请注意来源 https://blog.csdn.net/AI_LX/article/details/89669588

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
Created on Wed Dec 12 18:18:06 2018

@author: myhaspl
"""

import tensorflow as tf 
w1=tf.Variable(tf.random_normal([2,5],mean=1.0,stddev=1.0))
w2=tf.Variable(tf.random_normal([5,1],mean=1.0,stddev=1.0))
x=tf.constant([[0.9,2.8]])
y1=tf.matmul(x,w1)
y2=tf.matmul(y1,w2)

with  tf.Session() as ess:
    tf.global_variables_initializer().run()

    print(y2.eval())

[[16.396265]]