Method test_convert_to_cpu

test/external/external_test_example.py:28–32

(self, device)

class TestExample(unittest.TestCase):
  @multidevice_test
  def test_convert_to_cpu(self, device):
    a = Tensor([[1,2],[3,4]], device=device)
    assert a.numpy().shape == (2,2)
    b = a.to("CPU")
    assert b.numpy().shape == (2,2)

  @multidevice_test
  def test_2_plus_3(self, device):