with self.session(graph=ops.Graph(), use_gpu=True):
        self.assertAllClose(init_default(shape), np.eye(*shape))