| 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) |
|
|
