其他分享
首页 > 其他分享> > TensorFlow2.1中计算平方函数的tf.square()的用法

TensorFlow2.1中计算平方函数的tf.square()的用法

作者:互联网

功能:计算元素x里每一个元素的平方

tf.math.square
(
    x,
    name=None
)

 



参数说明:

x - 一个张量或者是稀疏张量。必须是下列类型之一:half, float32, float64, int32, int64, complex64, complex128.
name - 操作的名字 (可选参数).
示例如下:

import tensorflow as tf
a = tf.constant([1, 2, 3, 4])
b = tf.square(a)
print(b)
# tf.Tensor([ 1  4  9 16], shape=(4,), dtype=int32)

 


————————————————
版权声明:本文为CSDN博主「进击的Explorer」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/jpc20144055069/article/details/105110894

标签:int32,square,name,张量,TensorFlow2.1,tf,版权
来源: https://www.cnblogs.com/yibeimingyue/p/15563943.html