Pytorch call the torch function

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?


there isn’t a PyTorch way to call a torch layer.

1 Like

Got it !! Thanks ! Hope you can involve in the future. :stuck_out_tongue_winking_eye: