KMP library for tensors #300

Merged
grinisrit merged 215 commits from feature/tensor-algebra into dev 2021-05-08 09:48:04 +03:00
Showing only changes of commit a3ca861ebe - Show all commits

View File

@ -12,7 +12,7 @@ public interface ComplexTensorAlgebra<T,
public fun ComplexTensorType.angle(): RealTensorType public fun ComplexTensorType.angle(): RealTensorType
//https://pytorch.org/docs/stable/generated/torch.stft.html#torch.stft //https://pytorch.org/docs/stable/generated/torch.stft.html#torch.stft
public fun ComplexTensorType.stft( public fun ComplexTensorType.shortTimeFourierTransform(
nFFT: Int, nFFT: Int,
hopLength: Int, hopLength: Int,
winLength: Int, winLength: Int,
@ -22,7 +22,7 @@ public interface ComplexTensorAlgebra<T,
) )
//https://pytorch.org/docs/stable/generated/torch.istft.html#torch.istft //https://pytorch.org/docs/stable/generated/torch.istft.html#torch.istft
public fun ComplexTensorType.istft( public fun ComplexTensorType.inverseShortTimeFourierTransform(
nFFT: Int, nFFT: Int,
hopLength: Int, hopLength: Int,
winLength: Int, winLength: Int,