Pytorch call the torch function

vision
1

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,

2

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

1 Like
3

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