def _get_type(self, name): module = torch._import_dotted_name(self.data.__module__) return getattr(module, name)