作者:chen_h
微訊號 & QQ:862251340
微信公眾號:coderpai
我的部落格:請點選這裡
計劃現將 tensorflow 中的 Python API 做一個學習,這樣方便以後的學習。
原文連結
該章介紹有關數學符號操作的API
算術運算子
TensorFlow提供了一些操作,你可以使用基本的算術運算子新增到你的圖表。
tf.add(x, y, name = None)
解釋:這個函式返回x與y逐元素相加的結果。
注意:tf.add操作支援廣播形式,但是tf.add_n操作不支援廣播形式。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant(2)
b = tf.constant(3)
c = tf.add(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int8
,int16
,int32
,complex64
,int64
。y
: 一個Tensor
,資料型別必須和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.sub(x, y, name = None)
解釋:這個函式返回x與y逐元素相減的結果。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,2])
b = tf.constant(2)
c = tf.sub(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int8
,int16
,int32
,complex64
,int64
。y
: 一個Tensor
,資料型別必須和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.mul(x, y, name = None)
解釋:這個函式返回x與y逐元素相乘的結果。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,2])
b = tf.constant(2)
c = tf.mul(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int8
,int16
,int32
,complex64
,int64
。y
: 一個Tensor
,資料型別必須和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.div(x, y, name = None)
解釋:這個函式返回x與y逐元素相除的結果。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,2])
b = tf.constant(2)
c = tf.div(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int8
,int16
,int32
,complex64
,int64
。y
: 一個Tensor
,資料型別必須和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.mod(x, y, name = None)
解釋:這個函式返回x與y逐元素取餘的結果。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,2])
b = tf.constant(2)
c = tf.mod(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:int16
,int32
,float32
,float64
。y
: 一個Tensor
,資料型別必須和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
基礎的數學函式
TensorFlow提供了一些操作,你可以使用基本的數學函式,將它們新增到你的圖表。
tf.add_n(inputs, name = None)
解釋:這個函式的作用是對inputs列表中的元素相應位置累加。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,2], tf.int32)
b = tf.constant([3,4], tf.int32)
c = tf.add_n([a,b])
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
inputs
: 一個列表,其中至少有一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int64
,int32
,uint8
,int16
,int8
,complex64
,qint8
,quint8
,quint32
。並且列表中的每個Tensor
必須有相同的資料維度。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和inputs
相同。
tf.abs(x, name = None)
解釋:這個函式的作用返回x
的絕對值。
給定x
,它是一個實數Tensor
。這個操作返回一個tensor
,這個tensor
中的每個值是對應於x
中的每個值得絕對值。
如果,你需要處理複數的絕對值,那麼可以使用tf.complex_abs()
函式。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,-2])
c = tf.abs(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float
,double
,int64
或者int32
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和資料維度都和x
相同。
tf.neg(x, name = None)
解釋:這個函式的作用是得到x
中每個值得負數,即y = -x
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,-2])
c = tf.neg(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.sign(x, name = None)
解釋:這個函式是一個符號函式,按照如下規則轉換x
中的每一個值。
如果 x < 0,y = sign(x) = -1;
如果 x == 0,y = sign(x) = 0;
如果 x > 0,y = sign(x) = 1;
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([1,-2,0])
c = tf.sign(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int32
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.inv(x, name = None)
解釋:這個函式是計算x
中每個元素的倒數,即y = 1/x
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant(7.0)
c = tf.inv(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
譯者注:
我測試了一下這個API,但好像x
的資料型別只有是float
型別時才能成功。
tf.square(x, name = None)
解釋:這個函式是計算x
中每個元素的平方,即y = x*x = x^2
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.0,7.0])
c = tf.square(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.round(x, name = None)
解釋:這個函式是得到x
中每個元素離它最接近的整數。
比如:
# 'a' is [0.9, 2.5, 2.3, -4.4]
tf.round(a) ==> [ 1.0, 3.0, 2.0, -4.0 ]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.9, 0.0, -2.1, 2.0, 7.2])
c = tf.round(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是float
或者double
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同,資料維度和x
相同。
tf.sqrt(x, name = None)
解釋:這個函式是得到x
中每個元素的開平方值,即y = x^{1/2}
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2, 3], tf.float32)
c = tf.sqrt(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.rsqrt(x, name = None)
解釋:這個函式是得到x
中每個元素的開平方值的導數,即y = 1/x^{1/2}
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2, 3], tf.float32)
c = tf.rsqrt(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.pow(x, y, name = None)
解釋:這個函式是計算冪運算。
給定一個x
和y
,對應x
和y
中的每一個值,計算x^y
。
比如:
# tensor 'x' is [[2, 2]], [3, 3]]
# tensor 'y' is [[8, 16], [2, 3]]
tf.pow(x, y) ==> [[256, 65536], [9, 27]]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2, 3])
b = tf.constant([2, 3])
c = tf.pow(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float
,double
,int32
,complex64
,int64
。y
: 一個Tensor
,資料型別必須是以下之一:float
,double
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
。
tf.exp(x, name = None)
解釋:這個函式是計算x
中每個元素的指數,即y = e^x
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.0, 1], tf.float32)
c = tf.exp(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.log(x, name = None)
解釋:這個函式是計算x
中每個元素的自然對數,即y = log(x)
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.0, 1], tf.float32)
c = tf.log(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.ceil(x, name = None)
解釋:這個函式是返回不小於x
中每個元素的最小整數。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8], tf.float32)
c = tf.ceil(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.floor(x, name = None)
解釋:這個函式是返回不大於x
中每個元素的最大整數。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8], tf.float32)
c = tf.floor(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.maximum(x, y, name = None)
解釋:這個函式是逐個比較x
和y
中的值,求得最大值,即x > y ? x : y
。該函式支援廣播形式。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8, 0.0])
b = tf.constant(1.0)
c = tf.maximum(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,int64
。y
: 一個Tensor
,資料型別和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.minimum(x, y, name = None)
解釋:這個函式是逐個比較x
和y
中的值,求得最小值,即x < y ? x : y
。該函式支援廣播形式。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8, 0.0])
b = tf.constant(1.0)
c = tf.minimum(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,int64
。y
: 一個Tensor
,資料型別和x
相同。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.cos(x, name = None)
解釋:這個函式是計算x
中每個元素的餘弦值。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8, 0.0])
c = tf.cos(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
tf.sin(x, name = None)
解釋:這個函式是計算x
中每個元素的正弦值。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8, 0.0])
c = tf.sin(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,complex64
,int64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同。
矩陣數學函式
TensorFlow提供了一些操作,你可以使用新增基本的矩陣數學函式在你的圖中。
tf.diag(diagonal, name = None)
解釋:這個函式是給定一個對角值diagonal
,然後返回一個對角tensor
。
給定一個對角值diagonal
,這個操作返回一個對角tensor
,對角線上面的值是diagonal
,其餘值都用0
來填充。
假設diagonal
的維度為[D1, D2, ..., Dk]
,那麼輸出tensor
的秩為2k
,維度是[D1, D2, ..., Dk, D1, D2, ..., Dk]
,如下:
output[i1, i2, ..., ik, i1, i2, ..., ik] = diagonal[i1, .., ik],其餘值都是0。複製程式碼
比如:
# 'diagonal' is [1, 2, 3, 4]
tf.diag(diagonal) ==> [[1, 0, 0, 0]
[0, 2, 0, 0]
[0, 0, 3, 0]
[0, 0, 0, 4]]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
a = tf.constant([2.2, -1.8, 1.0])
c = tf.diag(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
diagonal
: 一個Tensor
,資料型別必須是以下之一:float32
,float64
,int32
,int64
。它的秩最大為3。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和diagonal
相同。
tf.transpose(a, perm = None, name = 'transpose')
解釋:將a
進行轉置,並且根據perm
引數重新排列輸出維度。
輸出資料tensor
的第i
維將根據perm[i]
指定。比如,如果perm
沒有給定,那麼預設是perm = [n-1, n-2, ..., 0]
,其中rank(a) = n
。預設情況下,對於二維輸入資料,其實就是常規的矩陣轉置操作。
比如:
input_data.dims = (1, 4, 3)
perm = [1, 2, 0]
# 因為 output_data.dims[0] = input_data.dims[ perm[0] ]
# 因為 output_data.dims[1] = input_data.dims[ perm[1] ]
# 因為 output_data.dims[2] = input_data.dims[ perm[2] ]
# 所以得到 output_data.dims = (4, 3, 1)
output_data.dims = (4, 3, 1)複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
sess = tf.Session()
input_data = tf.constant([[1,2,3],[4,5,6]])
print sess.run(tf.transpose(input_data))
print sess.run(input_data)
print sess.run(tf.transpose(input_data, perm=[1,0]))
input_data = tf.constant([[[1,2,3],[4,5,6],[7,8,9],[10,11,12]]])
print 'input_data shape: ', sess.run(tf.shape(input_data))
output_data = tf.transpose(input_data, perm=[1, 2, 0])
print 'output_data shape: ', sess.run(tf.shape(output_data))
print sess.run(output_data)
sess.close()複製程式碼
輸入引數:
a
: 一個Tensor
。perm
: 一個對於a
的維度的重排列組合。name
:(可選)為這個操作取一個名字。
輸出引數:
一個經過翻轉的
Tensor
。
tf.matmul(a, b, transpose_a = False, transpose_b = False, a_is_sparse = False, b_is_sparse = False, name = None)
解釋:將矩陣a
和矩陣b
進行相乘,得到矩陣a
*b
。
輸入資料必須是一個二維的矩陣,經過轉置或者不轉置,內部維度必須相匹配。
輸入矩陣必須有相同的資料型別,資料型別為:float
,double
,int32
,complex64
。
矩陣可以被設定為轉置操作,即transpose_a = True, transpose_b = True
。預設情況下,該標記都是被設定為False
。
如果矩陣中存在很多的0
,那麼我們可以使用sparse
標記,即a_is_sparse = True, b_is_sparse = True
。預設情況下,該標記都是被設定為False
。
比如:
# 2-D tensor `a`
a = tf.constant([1, 2, 3, 4, 5, 6], shape=[2, 3]) => [[1. 2. 3.]
[4. 5. 6.]]
# 2-D tensor `b`
b = tf.constant([7, 8, 9, 10, 11, 12], shape=[3, 2]) => [[7. 8.]
[9. 10.]
[11. 12.]]
c = tf.matmul(a, b) => [[58 64]
[139 154]]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(2,3))
b = tf.constant(np.random.rand(1,3))
c = tf.matmul(a, b, transpose_b = True)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
a
: 一個Tensor
,資料型別是float
,double
,int32
或者complex64
。b
: 一個Tensor
,資料型別和a
相同。transpose_a
: 如果該值維True
,那麼在矩陣計算之前,先將a
進行轉置。transpose_b
: 如果該值維True
,那麼在矩陣計算之前,先將b
進行轉置。a_is_sparse
: 如果該值維True
,那麼在矩陣計算的時候,將a
當做一個sparse
矩陣考慮。b_is_sparse
: 如果該值維True
,那麼在矩陣計算的時候,將b
當做一個sparse
矩陣考慮。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和a
相同。
tf.batch_matmul(x, y, adj_x = None, adj_y = None, name = None)
解釋:這個函式的作用是將兩個張量按批切片進行相乘。
將張量x
和y
進行切片(每個切片就是一個批的元素),然後將對應的x
和y
的每個切片進行相乘,將得到的結果按照原來批的大小進行重新安排。如果我們把adj_x
或者adj_y
設定成True
,在做乘法之前,每個獨立的切片可以組成它的共軛(其實相當於轉置)。
輸入的x
和y
是三維tensor
,或者更高維度的[..., r_x, c_x]
和[..., r_y, c_y]
。
輸出tensor
是一個三維的,或者更高維度的[..., r_o, c_o]
,其中:
r_o = c_x if adj_x else r_x
c_o = r_y if adj_y else c_y複製程式碼
計算過程如下:
out[..., :, :] = matrix(x[..., :, :]) * matrix(y[..., :, :])複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(2, 2, 3))
b = tf.constant(np.random.rand(3, 3, 1))
c = tf.batch_matmul(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(3, 2, 3, 1))
b = tf.constant(np.random.rand(3, 2, 3, 1))
c = tf.batch_matmul(a, b, adj_x = False, adj_y = True )
sess = tf.Session()
print sess.run(c)
print sess.run(tf.shape(c))
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是float32
,float64
,int32
或者complex64
,資料維度是三維的,或者更高維度[..., r_x, c_x]
。y
: 一個Tensor
,資料型別和x
相同,資料維度是三維的,或者更高維度[..., r_y, c_y]
。adj_x
: 這是一個可選的布林型別的值,預設情況下是False
。如果我們設定為True
,x
的每個切片將進行轉置。adj_y
: 這是一個可選的布林型別的值,預設情況下是False
。如果我們設定為True
,y
的每個切片將進行轉置。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和x
相同,資料維度是三維的,或者更高維度[..., r_o, c_o]
。
tf.matrix_determinant(input, name=None)
解釋:這個函式的作用是計算n
階矩陣的行列式。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(3, 3))
c = tf.matrix_determinant(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
input
: 一個Tensor
,資料型別是float32
或者float64
,資料維度是[M, M]
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
的標量,資料型別和input
相同。
tf.batch_matrix_determinant(input, name=None)
解釋:這個函式的作用是計算每個批(切片)的n
階矩陣的行列式。
輸入tensor
的資料維度必須是[..., M, M]
,其中內部必須是一個二維的方陣,對於所有的子矩陣,輸出結果是一個一維的tensor
。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(4, 2, 3, 3))
c = tf.batch_matrix_determinant(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
input
: 一個Tensor
,資料型別是float32
或者float64
,資料維度是[..., M, M]
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和input
相同,資料維度是[...]
。
tf.matrix_inverse(input, name=None)
解釋:這個函式的作用是計算n
階矩陣的逆矩陣,並且檢查可逆性。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(3, 3))
c = tf.matrix_inverse(a)
sess = tf.Session()
print sess.run(c)
d = tf.matmul(a, c)
print sess.run(d)
e = tf.matrix_determinant(d)
print sess.run(e)
sess.close()複製程式碼
輸入引數:
input
: 一個Tensor
,資料型別是float32
或者float64
,資料維度是[M, M]
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和input
相同,資料維度是[M, M]
。
tf.batch_matrix_inverse(input, name=None)
解釋:這個函式的作用是計算每個批(切片)的n
階矩陣的逆矩陣,並且檢查可逆性。
輸入tensor
的資料型別是[..., M, M]
,其中內部必須是一個二維的方陣,對於每一個子矩陣,輸出的矩陣的逆和輸入資料有相同的資料維度。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant(np.random.rand(2, 3, 3))
c = tf.batch_matrix_inverse(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
input
: 一個Tensor
,資料型別是float32
或者float64
,資料維度是[..., M, M]
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和input
相同,資料維度是[..., M, M]
。
tf.cholesky(input, name=None)
解釋:這個函式的作用是計算n
階矩陣的Cholesky分解。
輸入資料必須是一個對稱的正定矩陣,並且這個操作我們只會讀入矩陣的下三角部分,不會讀取矩陣的上三角部分。
輸出結果是經過Cholesky分解之後的一個對角線元素為正數的下三角實矩陣。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([[2, np.random.rand()], [-2, 5]], tf.float32)
c = tf.cholesky(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
input
: 一個Tensor
,資料型別是float32
或者float64
,資料維度是[M, M]
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和input
相同,資料維度是[M, M]
。
tf.batch_cholesky(input, name=None)
解釋:這個函式的作用是計算每個批(切片)的n
階矩陣的Cholesky分解。
輸入tensor
的資料型別是[..., M, M]
,其中內部必須是一個二維的方陣,並且滿足Cholesky分解的條件。輸出tensor
和輸入資料有相同的資料型別,並且每個切片都是經過Cholesky分解之後的值。
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([[[2, np.random.rand()], [-2, 5]], [[2, np.random.rand()], [-2, 5]]], tf.float32)
c = tf.batch_cholesky(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
input
: 一個Tensor
,資料型別是float32
或者float64
,資料維度是[..., M, M]
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別和input
相同,資料維度是[..., M, M]
。
複數函式
TensorFlow提供了一些複數函式,使得你可以去操作複數,你可以將它們新增到你的圖表。
tf.complex(real, imag, name=None)
解釋:這個函式的作用是將兩個實數轉換成一個複數。
這個操作就是去計算複數a + bj
,其中a
來自於輸入資料real
,表示實部,b
來自於輸入資料imag
,表示虛部。
輸入資料real
和imag
必須擁有相同的資料維度。
比如:
# tensor 'real' is [2.25, 3.25]
# tensor `imag` is [4.75, 5.75]
tf.complex(real, imag) ==> [[2.25 + 4.74j], [3.25 + 5.75j]]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([2.25, 3.25])
b = tf.constant([4.75, 5.75])
c = tf.complex(a, b)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
real
: 一個Tensor
,資料型別是float
。imag
: 一個Tensor
,資料型別是float
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別是complex64
。
tf.complex_abs(x, name=None)
解釋:這個函式的作用是計算複數的絕對值。
給定一個複數張量x
,這個操作是計算x
中的每個值的絕對值,並且返回一個float
型別的張量。在x
中的所有複數的形式必須是a + bj
的形式,那麼絕對值計算公式如下:
比如:
# tensor 'x' is [[-2.25 + 4.75j], [-3.25 + 5.75j]]
tf.complex_abs(x) ==> [5.25594902, 6.60492229]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([2.25 + 3.25j])
c = tf.complex_abs(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是complex64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別是float32
。
tf.conj(in_, name=None)
解釋:這個函式的作用是計算複數的複共軛。
給定一個複數張量in_
,這個操作是計算in_
中的每一個複數的複共軛。在in_
中所有複數的形式必須是a + bj
的形式,其中a
是實數部分,b
是虛數部分。
經過這個操作,複共軛的返回形式是a - bj
。
比如:
# tensor 'in' is [-2.25 + 4.75j, 3.25 + 5.75j]
tf.conj(in) ==> [-2.25 - 4.75j, 3.25 - 5.75j]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([2.25 + 3.25j])
c = tf.conj(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是complex64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別是complex64
。
tf.imag(in_, name=None)
解釋:這個函式的作用是返回複數的虛部。
給定一個複數張量in_
,這個操作是返回in_
中的每一個複數的虛部。在in_
中所有複數的形式必須是a + bj
的形式,其中a
是實數部分,b
是虛數部分。
比如:
# tensor 'in' is [-2.25 + 4.75j, 3.25 + 5.75j]
tf.imag(in) ==> [4.75, 5.75]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([2.25 + 3.25j])
c = tf.imag(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是complex64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別是float32
。
tf.real(in_, name=None)
解釋:這個函式的作用是返回複數的實部。
給定一個複數張量in_
,這個操作是返回in_
中的每一個複數的實部。在in_
中所有複數的形式必須是a + bj
的形式,其中a
是實數部分,b
是虛數部分。
比如:
# tensor 'in' is [-2.25 + 4.75j, 3.25 + 5.75j]
tf.real(in) ==> [-2.25, 3.25]複製程式碼
使用例子:
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import tensorflow as tf
import numpy as np
a = tf.constant([2.25 + 3.25j])
c = tf.real(a)
sess = tf.Session()
print sess.run(c)
sess.close()複製程式碼
輸入引數:
x
: 一個Tensor
,資料型別是complex64
。name
:(可選)為這個操作取一個名字。
輸出引數:
一個
Tensor
,資料型別是float32
。
CoderPai 是一個專注於演算法實戰的平臺,從基礎的演算法到人工智慧演算法都有設計。如果你對演算法實戰感興趣,請快快關注我們吧。加入AI實戰微信群,AI實戰QQ群,ACM演算法微信群,ACM演算法QQ群。詳情請關注 “CoderPai” 微訊號(coderpai)。