在keras 中获取张量 tensor 的维度大小实例
作者:skyfengye 时间:2023-08-27 21:58:12
在进行keras 网络计算时,有时候需要获取输入张量的维度来定义自己的层。但是由于keras是一个封闭的接口。因此在调用由于是张量不能直接用numpy 里的A.shape()。这样的形式来获取。这里需要调用一下keras 作为后端的方式来获取。当我们想要操作时第一时间就想到直接用 shape ()函数。其实keras 中真的有shape()这个函数。
shape(x)返回一个张量的符号shape,符号shape的意思是返回值本身也是一个tensor,
示例:
>>> from keras import backend as K
>>> tf_session = K.get_session()
>>> val = np.array([[1, 2], [3, 4]])
>>> kvar = K.variable(value=val)
>>> input = keras.backend.placeholder(shape=(2, 4, 5))
>>> K.shape(kvar)
<tf.Tensor 'Shape_8:0' shape=(2,) dtype=int32>
>>> K.shape(input)
<tf.Tensor 'Shape_9:0' shape=(3,) dtype=int32>
__To get integer shape (Instead, you can use K.int_shape(x))__
>>> K.shape(kvar).eval(session=tf_session)
array([2, 2], dtype=int32)
>>> K.shape(input).eval(session=tf_session)
array([2, 4, 5], dtype=int32)
如果直接调用这个出的不是我们想要的。我们想要的是tensor各个维度的大小。因此可以直接调用 int_shape(x) 函数。这个函数才是我们想要的。
>>> from keras import backend as K
>>> input = K.placeholder(shape=(2, 4, 5))
>>> K.int_shape(input)
(2, 4, 5)
>>> val = np.array([[1, 2], [3, 4]])
>>> kvar = K.variable(value=val)
>>> K.int_shape(kvar)
(2, 2)
最后这样我们就可以直接调用里面的大小。然后定义我们自己的keras 层了。
补充知识:获取Tensor的维度(x.shape和x.get_shape()的区别)
tf.shape(a)和a.get_shape()比较
相同点:都可以得到tensor a的尺寸
不同点:tf.shape()中a 数据的类型可以是tensor, list, array
a.get_shape()中a的数据类型只能是tensor,且返回的是一个元组(tuple)
import tensorflow as tf
import numpy as np
x=tf.constant([[1,2,3],[4,5,6]])
y=[[1,2,3],[4,5,6]]
z=np.arange(24).reshape([2,3,4])
sess=tf.Session()
# tf.shape()
x_shape=tf.shape(x) # x_shape 是一个tensor
y_shape=tf.shape(y) # <tf.Tensor 'Shape_2:0' shape=(2,) dtype=int32>
z_shape=tf.shape(z) # <tf.Tensor 'Shape_5:0' shape=(3,) dtype=int32>
print(sess.run(x_shape)) # 结果:[2 3]
print(sess.run(y_shape)) # 结果:[2 3]
print(sess.run(z_shape) ) # 结果:[2 3 4]
x_shape=x.get_shape()
print(x_shape) # 返回的是TensorShape([Dimension(2), Dimension(3)]),不能使用 sess.run() 因为返回的不是tensor 或string,而是元组 (2, 3)
x_shape=x.get_shape().as_list()
print(x_shape) # 可以使用 as_list()得到具体的尺寸,x_shape=[2 3] 这是重点 返回列表方便参加其他代码的运算
# y_shape=y.get_shape()
print(x_shape)# AttributeError: 'list' object has no attribute 'get_shape'
# z_shape=z.get_shape()
print(x_shape)# AttributeError: 'numpy.ndarray' object has no attribute 'get_shape' 或者a.shape.as_list()
来源:https://blog.csdn.net/nijiayan123/article/details/85780440
标签:keras,张量,tensor,维度
0
投稿
猜你喜欢
MySQL 数据编码 latin1 转 UTF8
2010-10-14 14:20:00
Python cookbook(数据结构与算法)实现查找两个字典相同点的方法
2022-07-20 22:09:46
Python中使用item()方法遍历字典的例子
2023-01-28 13:38:57
被jQuery折腾得半死,揭秘为何jQuery为何在IE/Firefox下均无法使用
2024-05-11 09:33:27
XML和XSLT结合使网站设计浑然一体
2008-09-05 17:13:00
python定义变量类型
2022-01-28 02:13:52
页面软键盘汉字输入
2010-09-01 20:41:00
详解Python中表达式i += x与i = i + x是否等价
2023-07-29 06:51:12
使用python socket分发大文件的实现方法
2022-05-23 10:45:40
MySQL Packet for query is too large 问题及解决方法
2024-01-29 07:55:52
Python+Appium新手教程
2022-12-01 04:12:44
在ASP.NET 2.0中操作数据之五十四:添加新记录时包含一个文件上传选项
2024-03-16 02:11:24
基于go手动写个转发代理服务的代码实现
2024-04-25 15:11:15
Python 新建文件夹与复制文件夹内所有内容的方法
2022-03-22 00:05:05
python中实现指定时间调用函数示例代码
2021-04-20 07:14:25
MSSQL报错:参数数据类型 text 对于 replace 函数的参数 1 无效的解决办法
2024-01-27 09:04:15
MySQL/MariaDB 如何实现数据透视表的示例代码
2024-01-13 16:12:18
Python中拆分具有多个分隔符的字符串方法实例
2022-07-01 00:38:20
python自然语言处理之字典树知识总结
2023-12-29 01:01:21
SQLServer行列互转实现思路(聚合函数)
2024-01-15 15:13:33