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次。
AtomGit 是由开放原子开源基金会联合 CSDN 等生态伙伴共同推出的新一代开源与人工智能协作平台。平台坚持“开放、中立、公益”的理念,把代码托管、模型共享、数据集托管、智能体开发体验和算力服务整合在一起,为开发者提供从开发、训练到部署的一站式体验。
更多推荐



所有评论(0)