python报错invalid argument_关于python：TensorFlow InvalidArgumentError / Value错误，代码变化很小...

代码：

import numpy as np

import tensorflow as tf

import pandas as pd

from sklearn.model_selection import train_test_split

x_data = np.linspace(0, 1000000, 1000)

y_true = np.sin(x_data)

y_true += np.random.randn(len(x_data))

feature_columns = [tf.feature_column.numeric_column('x', shape=[1])]

estimator = tf.estimator.DNNRegressor(feature_columns=feature_columns, hidden_units=[10,10,10], optimizer=lambda:

tf.train.AdamOptimizer(

learning_rate=0.1

))

X_train, X_test, y_train, y_test = train_test_split(x_data, y_true,    test_size=0.3)

input_function = tf.estimator.inputs.numpy_input_fn({'x': X_train},y_train,

batch_size=8,     num_epochs=None,

shuffle=True)

train_input_function = tf.estimator.inputs.numpy_input_fn({'x': X_train},y_train,

batch_size=8, num_epochs=1000,

shuffle=False)

test_input_function = tf.estimator.inputs.numpy_input_fn({'x': X_test},y_test,

batch_size=8, num_epochs=1000,

shuffle=False)

estimator.train(input_fn=input_function, steps=1000)

train_metrics = estimator.evaluate(input_fn=train_input_function, steps=1000)

test_metrics = estimator.evaluate(input_fn=test_input_function, steps=1000)

print('TRAINING DATA METRICS')

print(train_metrics)

print()

print('TEST DATA METRICS')

print(test_metrics)

print()

效果很好。

但是，如果我将y_true = np.sin(x_data)行更改为y_true=tf.square(x_data)，我会收到错误：

Traceback (most recent call last):

File"C:\Users\Admin\AppData\Local\Programs\Python\Python3