tensorflow-矩阵乘法

Posted 2021-01-26

#!/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]]