test_shape_inference.py 文件源码

def test_SimpleRNN(self):
        # all recurrent layers inherit output_shape
        # from the same base recurrent layer
        layer = SimpleRNN(2)
        input_data = np.random.random((2, 2, 3))
        check_layer_output_shape(layer, input_data)