1. 程式人生 > >tensorflow-矩陣乘法

tensorflow-矩陣乘法

code mean glob std lob asp ted orf pytho

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

tensorflow-矩陣乘法