Tensorflow loss function

I have a dumpy question I have a loss function that depends on Tensorflow and Keras backend, the question is can I use this loss function to train my PyTorch network.

Another question is there is any PyTorch function that equivalent to this TensorFlow command.
stft_true = tf.contrib.signal.stft(y_true,256,128,512,window_fn,pad_end=False)



Have you tried torch.stft that looks like the same thing.

Yes, @albanD
It seems the same function.