def METHOD_NAME(test_case, device):
    np_dim = np.array([[1, 2, 3], [1, 2, 3]], dtype=int)
    flow_dim = flow.tensor(np_dim).to(device)
    torch_dim = torch.tensor(np_dim).to(device)
    np_random_array = np.random.randn(2, 3, 4, 5)
    flow_tensor = flow.tensor(np_random_array).to(device)
    torch_tensor = torch.tensor(np_random_array).to(device)
    flow_result = flow.tensordot(flow_tensor, flow_tensor, dims=flow_dim)
    torch_result = torch.tensordot(torch_tensor, torch_tensor, dims=torch_dim)
    test_case.assertTrue(
        np.allclose(
            flow_result.numpy(),
            torch_result.cpu().numpy(),
            rtol=0.0001,
            atol=0.0001,
        )
    )