Fix warnings for newer PyTorch versions #30
Description
This includes correct use of softmax() which now expects an explicit dimension, and tensor copies in the Exporter.
{{ message }}
This includes correct use of softmax() which now expects an explicit dimension, and tensor copies in the Exporter.