@@ -317,0 +356,4 @@
+        input_ms = cast_to_ms_tensor(self)
+        return cast_to_adapter_tensor(input_ms.bmm(batch2))
+
+    def clamp(self, min, max):

@@ -317,0 +416,4 @@
+    def sub(self, other, *, alpha=1):
+        input_ms = cast_to_ms_tensor(self)
+        input_other = cast_to_ms_tensor(other) * alpha
+        output = ms.Tensor.sub(input_ms, input_other)