时间:2021-05-22
实验发现,tensorflow的tensor张量的shape不支持直接作为tf.max_pool的参数,比如下面这种情况(一个错误的示范):
self.max_pooling1 = tf.nn.max_pool(self.l_6, ksize = [1, tf.shape(self.F1)[0], 1, 1], strides = [1, 1, 1, 1], padding = 'VALID', name = 'maxpool1')我在max_pool的过程中想对特征每一列进行max_pooling,但因为tensor F1张量没有值,而ksize此处需要一个int类型的常量,所以tf.shape(self.F1)[0]无法作为ksize的参数。
一些人的做法是固定行数,比如pointnet固定为2048个点,但这样做需要重新采样,过于麻烦。
而既然tensorflow不提供标准的max_pool层实现,其实可以自己实现每一列取最大值的操作,充当自己的max_pool。如下图所示:
def max_pooling(self, NC): out = tf.reduce_max(NC, reduction_indices=[0]) return out利用reduce_max刚好能实现这个操作,其中reduction_indices可以指定取哪一维的最大值。这个函数的详细功能可以参考官方文档。
这种情况和tf.reshape不同,tf.reshape是可以使用tf.shape(tensor)[i]的,可能两者对参数的要求不一样。
以上这篇tensorflow之获取tensor的shape作为max_pool的ksize实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。
初学tensorflow,如果写的不对的,请更正,谢谢!tf.reshape(tensor,shape,name=None)函数的作用是将tensor变换为参数
#tf.shape(tensor)和tensor.shape()的区别a=tf.zeros([4,5,4,5,6])print(type(a.shape))pr
tensorflow版本1.4获取变量维度是一个使用频繁的操作,在tensorflow中获取变量维度主要用到的操作有以下三种:Tensor.shapeTenso
涉及随机数以及类RNN的网络构建常常需要根据输入shape,决定中间变量的shape或步长。tf.shape函数不同于tensor.shape.as_list(
如下所示:tf.constant(value,dtype=None,shape=None)创建一个常量tensor,按照给出value来赋值,可以用shape来