Source code for draugr.torch_utilities.tensors.mixins
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import torch
from draugr.torch_utilities.tensors import to_tensor
__author__ = "Christian Heider Nielsen"
__all__ = ["TensoriseMixin"]
[docs]class TensoriseMixin(object):
"""
Tensorise attributes at set"""
device = "cpu"
dtype = torch.float # Default values may be monkey patched for other types
def __setattr__(self, key, value):
super().__setattr__(key, to_tensor(value, dtype=self.dtype, device=self.device))