class torch.nn.CrossEntropyLoss(weight=None, size_average=True)[source] · input : 包含每个类的得分,2-D tensor,shape为batch*n · target: 大小为n 的1—D tensor,包含 ...
確定! 回上一頁