tensorflow之tf.square()

tensorflow之tf.square()

import tensorflow as tf
a=tf.constant([1,2,3,4])
b=tf.square(a)
with tf.Session() as sess:
    print(sess.run(b))

tf.square是对a里面的元素求平方

返回笔记列表
入门小站