对tensorflow 中tile函数的使用详解

tensorflow中tile是用来复制tensor的指定维度,具体看下面的代码:

import tensorflow as tf
 
a = tf.constant([[1, 2], [3, 4], [5, 6]], dtype=tf.float32)
 
a1 = tf.tile(a, [2, 2])
with tf.Session() as sess:
  print(sess.run(a1))

结果就是:

[[ 1. 2. 1. 2.]
 [ 3. 4. 3. 4.]
 [ 5. 6. 5. 6.]
 [ 1. 2. 1. 2.]
 [ 3. 4. 3. 4.]

 [ 5. 6. 5. 6.]]

因为

a1 = tf.tile(a, [2, 2]) 表示把a的第一个维度复制两次,第二个维度复制2次。

以上这篇对tensorflow 中tile函数的使用详解就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持脚本之家。

你可能感兴趣的:(对tensorflow 中tile函数的使用详解)