如果reduction='none',即不进行任何处理,则为(N,*),形状与输入相同。 举例:. loss = nn.L1Loss() input = torch.randn(1, 2, requires_grad= ...
確定! 回上一頁