https://zh-v2.d2l.ai/chapter_multilayer-perceptrons/mlp-scratch.html
你应该使用torch.max而不是max
函数原型：
torch.max(input, other, out=None) → Tensor
参数：
input (Tensor) – 输入张量
other (Tensor) – 输出张量
out (Tensor, optional) – 结果张量
例子：
a = torch.randn(4)
a
1.3869
0.3912
-0.8634
-0.5468
[torch.FloatTensor of size 4]
b = torch.randn(4)
b
1.0067
-0.8010
0.6258
0.3627
[torch.FloatTensor of size 4]
torch.max(a, b)
1.3869
0.3912
0.6258
0.3627
[torch.FloatTensor of size 4]