tf.square()、tf.pow()、tf.math.pow()函数介绍和示例

tf.square()、tf.pow()、tf.math.pow()函数介绍和示例

1. tf.square()

释义：对各元素求平方

示例：

import tensorflow as tf

X = tf.constant([1, 2, 3, 4], dtype=tf.float32, name=None)
Y = tf.square(X)         # 平方操作

with tf.Session() as sess:
    print(sess.run(Y))

[ 1.  4.  9. 16.]

2. tf.pow(x, y, name=None)

释义：幂值计算操作，即 x^y

示例：

import tensorflow as tf

x = tf.constant(2, dtype=tf.float32)
y = tf.constant(3, dtype=tf.float32)
z = tf.pow(x, y)                              # 标量^标量操作

X1 = tf.constant(3, dtype=tf.float32)
Y1 = tf.constant([1, 2, 3], dtype=tf.float32)        
Z1 = tf.pow(X1, Y1)                           # 标量^向量操作（矩阵同理）

X2 = tf.constant([[1, 2, 3], [4, 5, 6]], dtype=tf.float32)
Y2 = tf.constant([[2,2,2],[3,3,3]], dtype=tf.float32)
Z2 = tf.pow(X2, Y2)                           # 矩阵^矩阵操作（向量同理），对应位置的值进行幂值计算

with tf.Session() as sess:
    print(sess.run(z))
    print('='*30)
    
    print(sess.run(Z1))
    print('='*30)
    
    print(sess.run(Z2))

8.0
==============================
[ 3.  9. 27.]
==============================
[[  1.   4.   9.]
 [ 64. 125. 216.]]

3. tf.math.pow(x, y, name=None)

释义：同 tf.pow(x, y, name=None)

示例：

import tensorflow as tf

x = tf.constant(2, dtype=tf.float32)
y = tf.constant(3, dtype=tf.float32)
z = tf.math.pow(x, y)                              # 标量^标量操作

X1 = tf.constant(3, dtype=tf.float32)
Y1 = tf.constant([1, 2, 3], dtype=tf.float32)        
Z1 = tf.math.pow(X1, Y1)                           # 标量^向量操作（矩阵同理）

X2 = tf.constant([[1, 2, 3], [4, 5, 6]], dtype=tf.float32)
Y2 = tf.constant([[2,2,2],[3,3,3]], dtype=tf.float32)
Z2 = tf.math.pow(X2, Y2)                           # 矩阵^矩阵操作（向量同理），对应位置的值进行幂值计算

with tf.Session() as sess:
    print(sess.run(z))
    print('='*30)
    
    print(sess.run(Z1))
    print('='*30)
    
    print(sess.run(Z2))

8.0
==============================
[ 3.  9. 27.]
==============================
[[  1.   4.   9.]
 [ 64. 125. 216.]]