tensorflow python3 tf_export.py functools.partial 在类上和在函数上的用法

博客介绍了functools.partial在类和函数上的应用。作者在阅读tf_export.py源码时，发现functools.partial参数为类名，而网上资料多讲参数为函数的情况。还提到若要减少函数参数个数，可用partial()函数固定参数值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

functools.partial 在类上的demo.

from functools import partial
class api_export(object):  # pylint: disable=invalid-name
  """Provides ways to export symbols to the TensorFlow API."""

  def __init__(self, *args, **kwargs):  # pylint: disable=g-doc-args
    self._names = args
    self._names_v1 = kwargs.get('v1', args)
    if 'v2' in kwargs:
      raise ValueError('You passed a "v2" argument to tf_export. This is not '
                       'what you want. Pass v2 names directly as positional '
                       'arguments instead.')
    self._api_name = kwargs.get('api_name', 'TENSORFLOW_API_NAME')
    self._overrides = kwargs.get('overrides', [])
    self._allow_multiple_exports = kwargs.get('allow_multiple_exports', False)

    #self._validate_symbol_names()


tf_export = partial(api_export, api_name='TENSORFLOW_API_NAME')

estimator_export = partial(api_export, api_name='ESTIMATOR_API_NAME')

keras_export = partial(api_export, api_name='KERAS_API_NAME')

print(tf_export()._api_name)
print(estimator_export()._api_name)
print(keras_export()._api_name)

functools.partial 的用法

今天读到tf_export.py的源码。

functools.partial 的参数是类名。查看网上的资料大多数都在讲参数是函数的。

如果需要减少某个函数的参数个数，你可以使用 functools.partial() 。partial()

函数允许你给一个或多个参数设置固定的值，减少接下来被调用时的参数个数。

使用 partial() 函数来固定某些参数值

functools.partial 在函数上的demo.

from functools import partial
def mfun(a,b,c):
	print(a,b,c)


parfun1 = partial(mfun, 3,5)# a = 3, b = 5

parfun2 = partial(mfun, c=33)# c=33

parfun1(99)
parfun2(44,66)

运行结果