出现这种情况的时候说明使用label出错
下面展示一些
内联代码片
。
// A code block
import torch
import torch.nn.functional as F
a = torch.Tensor([0,1])
b = torch.Tensor([0.1,0.3,0.4],[0.2,0.5,0.8])
print(F.cross_entropy(input = b,target=a))
结果就会出现上面报错
// An highlighted block
import torch
import torch.nn.functional as F
a = torch.Tensor([0,1])
b = torch.Tensor([0.1,0.3,0.4],[0.2,0.5,0.8])
print(F.cross_entropy(input = b,target=a))
结果就会出现上面报错
正确的代码是
下面展示一些 内联代码片
。
// A code block
import torch
import torch.nn.functional as F
a = torch.Tensor([0,1]).long()
b = torch.Tensor([0.1,0.3,0.4],[0.2,0.5,0.8])
print(F.cross_entropy(input = b,target=a));
// An highlighted block
import torch
import torch.nn.functional as F
a = torch.Tensor([0,1]).long()
b = torch.Tensor([0.1,0.3,0.4],[0.2,0.5,0.8])
print(F.cross_entropy(input = b,target=a))
这样就整好了