Tensorflow loss function

audio
1

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)

Thanks

2

Hey,

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

3

Yes, @albanD
It seems the same function.