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?
there isn’t a PyTorch way to call a torch layer.
Got it !! Thanks ! Hope you can involve in the future.