首页 > 其他 > 详细

Modify tensor shape

时间:2020-06-08 17:18:35      阅读:34      评论:0      收藏:0      [点我收藏+]

一 Transformation method

  • TensorFlow tensor has two shape transfomations,dynamic shape and static shape.Static shape conversion calls set_shape().The conversion must be the same shape as the initial tensor creation.For a tensor with a static shape fixed,the static shape cannot be set again.
  • Dynamic shape conversion calls tf.reshape(tensor,shape),the number of elements of the dynamically transformed shape tensor must be the same as before modification.

二 Code example

import os
os.environ[‘TF_CPP_MIN_LOG_LEVEL‘] = ‘2‘
import tensorflow as tf 
def tensor_shape_demo():
    a_p = tf.placeholder(dtype=tf.float32,shape=[None,None])
    b_p = tf.placeholder(dtype=tf.float32,shape=[None,5])
    print(‘======================Static shape modification‘)
    print(‘The shape before modification:‘)
    print(‘a_p:‘,a_p)
    print(‘b_p:‘,b_p)
    print(‘The shape after modification:‘)
    a_p.set_shape([3,4])
    b_p.set_shape([2,5])
    print(‘a_p:‘,a_p)
    print(‘b_p:‘,b_p)
    print(‘======================Dynamic shape modification‘)
    c_p = tf.placeholder(dtype=tf.float32,shape=[3,4])
    print(‘The shape before modification:‘)
    print(‘c_p:‘,c_p)
    print(‘The shape after modification:‘)
    c_p = tf.reshape(c_p,shape=[2,6])
    print(‘c_p:‘,c_p)
    c_p = tf.reshape(c_p,shape=[4,3])
    print(‘c_p:‘,c_p)
    c_p = tf.reshape(c_p,shape=[3,4,1])
    print(‘c_p:‘,c_p)
if __name__ == ‘__main__‘:
    tensor_shape_demo()

三 Operation result

技术分享图片

Modify tensor shape

原文:https://www.cnblogs.com/monsterhy123/p/13066715.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!