def test_variable_shape(self):
        """Test retrieving the shape of a variable."""
        v = backend.Variable(initializer=np.array([[1.0, 2.0], [3.0, 4.0]]))
        self.assertEqual(v.shape, (2, 2))
Test retrieving the shape of a variable.
keras/src/backend/common/variables_test.py