to

40fcc1e1 · Benjamin Thomas Graham · ecc6e480 · 40fcc1e1
Commit 40fcc1e1 authored Mar 15, 2019 by Benjamin Thomas Graham
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

sparseconvnet/sparseConvNetTensor.py sparseconvnet/sparseConvNetTensor.py +4 -0

No files found.
--- a/sparseconvnet/sparseConvNetTensor.py
+++ b/sparseconvnet/sparseConvNetTensor.py
@@ -22,6 +22,10 @@ class SparseConvNetTensor(object):
        t = self.metadata.getSpatialLocations(spatial_size)
        return t

+    def to(self, device):
+        self.features=self.features.to(device)
+        return self
+
    def type(self, t=None):
        if t:
            self.features = self.features.type(t)