nn.Linear

relu?

 

concisely

• inherit from nn.Module
• init layer in __init__
• implement forward()

step 1

 

step 2

 

step 3

 

nn.ReLU v.s. F.relu()

• class-style API
  • 因此需要先实例化
• function-style API
  • 因此可以直接传参

 

Train