paddle定义了一个简单网络，如何用paddle.summary查看网络结构？？

Question

paddle定义了一个简单网络，如何用paddle.summary查看网络结构？？

class LSTM(paddle.nn.Layer):
    def __init__(self, input_size=1, hidden_size=16): 
        super().__init__()
        self.rnn = paddle.nn.LSTM(input_size=input_size, hidden_size=hidden_size,  num_layers=3)
        self.linear = paddle.nn.Linear(hidden_size, 1)

    def forward(self, inputs):
        y, (hidden, cell) = self.rnn(inputs)
        output = self.linear(hidden[-1])
   
     
        return output         

Answer 1

import paddle

class LSTM(paddle.nn.Layer):
    def __init__(self, input_size=1, hidden_size=16):
        super().__init__()
        self.rnn = paddle.nn.LSTM(input_size=input_size, hidden_size=hidden_size, num_layers=3)
        self.linear = paddle.nn.Linear(hidden_size, 1)

    def forward(self, inputs):
        y, (hidden, cell) = self.rnn(inputs)
        output = self.linear(hidden[-1])
        return output

input_size = 12  # 任意可修改
time_steps = 2  # 任意可修改
model = LSTM(input_size)
summary = paddle.summary(model, (None, time_steps, input_size))
print(summary)