动手学深度学习 softmax回归简洁实现报错
在网上查询了很多源码
运行结果报错
请问这是哪里出问题了?
动手学深度学习 softmax回归简洁实现报错
在网上查询了很多源码
【相关推荐】
def softmax(X):# 将输出值变换成值为正且和为1的概率分布
X_exp = X.exp()
partition = X_exp.sum(dim=1, keepdim=True)
return X_exp / partition # 这里应用了广播机制
X = torch.rand((2, 5))
X_prob = softmax(X)
print(X_prob, X_prob.sum(dim=1))# tensor([[0.3089, 0.1828, 0.1810, 0.1556, 0.1716], [0.1133, 0.2303, 0.1293, 0.2722, 0.2549]]) tensor([1., 1.])