mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
22 lines
675 B
Python
22 lines
675 B
Python
from torch._C import _set_backcompat_broadcast_warn
|
|
from torch._C import _get_backcompat_broadcast_warn
|
|
from torch._C import _set_backcompat_keepdim_warn
|
|
from torch._C import _get_backcompat_keepdim_warn
|
|
|
|
|
|
class Warning(object):
|
|
def __init__(self, setter, getter):
|
|
self.setter = setter
|
|
self.getter = getter
|
|
|
|
def set_enabled(self, value):
|
|
self.setter(value)
|
|
|
|
def get_enabled(self):
|
|
return self.getter()
|
|
|
|
enabled = property(get_enabled, set_enabled)
|
|
|
|
broadcast_warning = Warning(_set_backcompat_broadcast_warn, _get_backcompat_broadcast_warn)
|
|
keepdim_warning = Warning(_set_backcompat_keepdim_warn, _get_backcompat_keepdim_warn)
|