Added various PyTorch wrapper nodes to generate tensors.

2025-02-27 13:05:17 -08:00
parent 8e36913138
commit b0782fb422
34 changed files with 1382 additions and 1 deletions
@@ -0,0 +1,14 @@
+import torch
+
+DTYPE_MAPPING = {
+    "float32": torch.float32,
+    "float16": torch.float16,
+    "bfloat16": torch.bfloat16,
+    "float64": torch.float64,
+    "uint8": torch.uint8,
+    "int8": torch.int8,
+    "int16": torch.int16,
+    "int32": torch.int32,
+    "int64": torch.int64,
+    "bool": torch.bool
+}