tf.sqrt()函数
时间: 2024-06-04 18:04:03 浏览: 221
matlab.rar_matlab函数大全
tf.sqrt()函数是TensorFlow中的一个数学运算函数,用于计算输入张量的平方根。例如,如果输入一个标量或者一个形状为[batch_size, num_features]的张量,那么输出就是一个与输入形状相同的张量,其中每个元素都是输入元素的平方根。
下面是tf.sqrt()函数的示例代码:
```
import tensorflow as tf
x = tf.constant([4.0, 9.0, 16.0])
y = tf.sqrt(x)
print(y)
```
输出结果为:
```
tf.Tensor([2. 3. 4.], shape=(3,), dtype=float32)
```
阅读全文