You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
x shape is (500|32 )
val yx = torch.sum(x,new DimnameArrayRef(1))
println(s"FeaturesLinear x sum shape ${x.sum().shape().mkString("|")} two ${x.sum(new DimnameArrayRef(1)).shape().mkString("|")} x shape ${x.shape().mkString("|")} sum yx shape ${yx.shape().mkString("|")}")
in console
FeaturesLinear x sum shape two x shape 500|32 sum yx shape
in old pytorch-javacpp version, I remember the sum maybe can really work, why new version get the bad result ,is DimnameArrayRef class I use it with false operate?