Change test, because it was building an incorrect graph.
It makes no sense to have a softmax_grad node with inputs that are not
matrices of the same size. C implementation would have complained.
T_SoftmaxGrad.test_infer_shape now passes in FAST_COMPILE.
正在显示
请
注册
或者
登录
后发表评论