level 1
荷包蛋丶😁
楼主
我想问下为什么tf.py_function()中的自己定义的function方法没有被调用是怎么回事,程序也没有报错。
def Myshow_all(self,inputs):
def showTensor(inputs):
a = inputs.numpy()
print(a)
return a
y=tf.py_function(func=showTensor,inp=[inputs],Tout=tf.float32)
print(y.shape)
return y
def _pooling_function(self, inputs, pool_size, strides,
padding, data_format):
output = K.pool2d(inputs, pool_size, strides,
padding, data_format,
pool_mode='max')
a=self.Myshow_all(inputs)
print(a)
print(type(a))
print(a.shape)
return output
这个是相关代码,本意是想将Tensor类型转为numpy类型,观察一下矩阵里面具体的值。
2020年09月23日 10点09分
1
def Myshow_all(self,inputs):
def showTensor(inputs):
a = inputs.numpy()
print(a)
return a
y=tf.py_function(func=showTensor,inp=[inputs],Tout=tf.float32)
print(y.shape)
return y
def _pooling_function(self, inputs, pool_size, strides,
padding, data_format):
output = K.pool2d(inputs, pool_size, strides,
padding, data_format,
pool_mode='max')
a=self.Myshow_all(inputs)
print(a)
print(type(a))
print(a.shape)
return output
这个是相关代码,本意是想将Tensor类型转为numpy类型,观察一下矩阵里面具体的值。