© 1999-2048 dssz.net 粤ICP备11031372号
[其它] keras获得model中某一层的某一个Tensor的输出维度教程
说明:获得某层tensor的输出维度 代码如下所示: from keras import backend as K wraps(Conv2D) def my_conv(*args,**kwargs): new_kwargs={'kernel_regularizer':l2(5e-6)} new_kwargs['padding']='valid' #'same' new_kwargs['strides']=(2,2) if kwargs.get('strides')==(2,2) else<weixin_38582506> 上传 | 大小:37kb