pytorch1.2中的使用自己定义的交叉熵时报错

官方定义的:

img


我自己写的:

img


然后我用自己的替换官方的,报错

img


我就纳闷了,咋回事呢?该怎么改才能让我的这串代码和官方的等价,能跑起来就行,
是不是还得需要定义一个forward()前向传播的函数啊,我是菜鸡,怎么改?

你定义好了这个函数,就要写段代码去调用啊,而不是直接运行这个,调用按照格式来就是了

input = #自己写
target = #自己写
Crossentropy(input, target)