當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python tensorflow.math.rsqrt()用法及代碼示例

TensorFlow是Google設計的開源python庫,用於開發機器學習模型和深度學習神經網絡。 rsqrt()用於查找x的平方根的逐元素倒數。

用法:tf.math.rsqrt(x, name)

參數:

  • x:它是輸入張量。此張量的允許dtype是bfloat16,half,float32,float64。
  • name(optional):它定義了操作的名稱。

返回值:
它返回與x相同dtype的張量。

範例1:



Python3

# 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:

Python3

# 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)




注:本文由純淨天空篩選整理自 Python – tensorflow.math.rsqrt()。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。