torch.nn.Linear和torch.nn.MSELoss

1.torch.nn.Linear------計算y=ax+b in_features爲輸入維度 out_features爲輸出維度 bias爲b的值 2.torch.nn.MSELoss size_average爲是最後loss加起來過後是否除以N reduce----因爲loss爲張量,reduce爲true表示將所有的lossi累加達到降維的目的
相關文章
相關標籤/搜索