lib/torch.rb in torch-rb-0.11.0 vs lib/torch.rb in torch-rb-0.11.1
- old
+ new
@@ -420,21 +420,9 @@
data = data.map { |v| v ? 1 : 0 } if options[:dtype] == :bool
_tensor(data, size, tensor_options(**options))
end
- # center option
- def stft(input, n_fft, hop_length: nil, win_length: nil, window: nil, center: true, pad_mode: "reflect", normalized: false, onesided: true, return_complex: nil)
- if center
- signal_dim = input.dim
- extended_shape = [1] * (3 - signal_dim) + input.size
- pad = n_fft.div(2).to_i
- input = NN::F.pad(input.view(extended_shape), [pad, pad], mode: pad_mode)
- input = input.view(input.shape[-signal_dim..-1])
- end
- _stft(input, n_fft, hop_length, win_length, window, normalized, onesided, return_complex)
- end
-
private
def to_ivalue(obj)
case obj
when String