chore: import upstream snapshot with attribution

python/dgl/storages/tensor.py

@@ -0,0 +1,17 @@
+"""Feature storages for tensors across different frameworks."""
+from .. import backend as F
+from .base import FeatureStorage
+
+
+class BaseTensorStorage(FeatureStorage):
+    """FeatureStorage that synchronously slices features from a tensor and transfers
+    it to the given device.
+    """
+
+    def __init__(self, tensor):
+        self.storage = tensor
+
+    def fetch(
+        self, indices, device, pin_memory=False, **kwargs
+    ):  # pylint: disable=unused-argument
+        return F.copy_to(F.gather_row(self.storage, indices), device, **kwargs)