Hi All,
I aim to use the cudnn.VolumetricCrossEntropyCriterion() layer in the pytorch, but I seach the document and cannot find me that is equal to this layer.
Btw, is there any way to use pytorch to call the torch layer?
Best,