tensorflow1.5 多元线性模型

# -*- coding: utf-8 -*-
"""
@author: maiobinfei
"""

import os
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
learn_rate=0.2
train_epochs=301
tfx=np.float32(np.random.rand(2,100))
tfy=np.dot(np.float32([0.1,0.2]),tfx)+0.3
fig=plt.figure()
ax=fig.add_subplot(111,projection='3d')
ax.scatter(tfx[0][:99],tfx[1][:99],tfy[:99],c='r')
plt.show()

x=tf.placeholder(tf.float32,[None,None])
y=tf.placeholder(tf.float32,[None,None])
w=tf.Variable(tf.random_uniform([1,2],-1.0,1.0))
b=tf.Variable(tf.zeros([1]))
y=tf.matmul(w,tfx)+b
loss=tf.reduce_mean(tf.square(y-tfy))
optimizer=tf.train.GradientDescentOptimizer(learn_rate)
train=optimizer.minimize(loss)
init=tf.g