def call(self, inputs, **kwargs):
return tf.space_to_depth(inputs, self.block_size)
微信
公众号
扫码关注公众号
微信
公众号