极客教程Python – tensorflow.math.rsqrt()
TensorFlow是谷歌设计的开源python库，用于开发机器学习模型和深度学习神经网络。rsqrt()用于查找x的平方根的元素倒数。
语法: tf.math.rsqrt(x, name)
参数 :
- x:它是输入的张量。这个张量允许的dtype是bfloat16, half, float32, float64。
- name（可选）:它定义了操作的名称。
返回值:
它返回一个与x具有相同dtype的张量。
示例 1:
# Importing the library
import tensorflow as tf
# Initializing the input tensor
a = tf.constant([.2, 1.6, 3.7, 1], dtype = tf.float64)
# Printing the input tensor
print('a: ', a)
# Calculating result
res = tf.math.rsqrt(a)
# Printing the result
print('Result: ', res)
输出:
a: tf.Tensor([0.2 1.6 3.7 1. ], shape=(4, ), dtype=float64)
Result: tf.Tensor([2.23606798 0.79056942 0.51987524 1. ], shape=(4, ), dtype=float64)
示例 2:
# importing the library
import tensorflow as tf
# Initializing the input tensor
a = tf.constant([2.5, 5.5, 1.5, 6.5], dtype = tf.float64)
# Printing the input tensor
print('a: ', a)
# Calculating result
res = tf.math.rsqrt(a)
# Printing the result
print('Result: ', res)
输出:
a: tf.Tensor([2.5 5.5 1.5 6.5], shape=(4, ), dtype=float64)
Result: tf.Tensor([0.63245553 0.42640143 0.81649658 0.39223227], shape=(4, ), dtype=float64)