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次。
新一代开源开发者平台 GitCode,通过集成代码托管服务、代码仓库以及可信赖的开源组件库,让开发者可以在云端进行代码托管和开发。旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。
更多推荐


所有评论(0)