tensorflow.shape() 函数用法

# tf.shape()  返回的是tensor的维度

代码如下:


import tensorflow as tf     #导入tensorflow包

# 定义一个tensor,是由2行4列构成 类型为:float32

constant_tensor=tf.constant(value=[[1.1,2.2,3.3,4.4],[4.1,3.1,2.1,1.1]], dtype=tf.float32)

# shape函数是现实tensor的维度结构

shape_constant_tensor=tf.shape(constant_tensor)

# 定义一个Session

sess=tf.Session()

# 运行这个Session

sess.run(shape_constant_tensor)


输出结果:

array([2, 4]


恭喜您学习完成!

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。