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