@@ -0,0 +22,4 @@
+    ms_tensor_1 = ms_torch.Tensor(2, 3, 4)
+    torch_tensor_1 = torch.Tensor(2, 3, 4)
+
+    assert ms_tensor_1.size().numel() == torch_tensor_1.size().numel()

@@ -0,0 +46,4 @@
+    msa_net = TestNet(size_func)
+    ms_tensor = ms_torch.Tensor(2, 3)
+    output = msa_net(ms_tensor)
+    assert str(output.size()) == "mindtorch.Size([2, 3])"

@@ -291,6 +292,8 @@ class Tensor(StubTensor, metaclass=_TensorMeta):
         if len(data) == 1:
             if isinstance(data[0], (int, np.integer)):
                 _shape = data
+            elif isinstance(data[0], Size):