tensorflow中的函数

 
 

一: tf.nn.l2_loss(tensor) = sum(tensor ^ 2)/2

import tensorflow as tf

a=tf.constant([1,2,3],dtype=tf.float32)
b=tf.constant([[1,1],[2,2],[3,3]],dtype=tf.float32)

with tf.Session() as sess:
    print('a:')
    print(sess.run(a))
    print('a_loss:')
    print(sess.run(tf.nn.l2_loss(a)))
    print('b:')
    print(sess.run(b))
    print('b_loss:')
    print(sess.run(tf.nn.l2_loss(b)))
    sess.close()



猜你喜欢

转载自blog.csdn.net/zhangxinyan00/article/details/79467561