experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #42269 from WindQAQ:fix-stft-frame

Browse Source 
PiperOrigin-RevId: 326366725
Change-Id: I60b77331b0f1e922588421a949c0d61e8118fe50
This commit is contained in:
TensorFlower Gardener 2020-08-12 20:01:17 -07:00
commit fb3bf4642d
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/python/ops/signal/shape_ops.py
View File

@ -167,11 +167,11 @@ def frame(signal, frame_length, frame_step, pad_end=False, pad_value=0, axis=-1,
0, frame_length + frame_step * (num_frames - 1) - length_samples) 0, frame_length + frame_step * (num_frames - 1) - length_samples)
# Pad the inner dimension of signal by pad_samples. # Pad the inner dimension of signal by pad_samples.
paddings = array_ops.concat( paddings = array_ops.concat([
[array_ops.zeros([num_outer_dimensions, 2], dtype=pad_samples.dtype), array_ops.zeros([num_outer_dimensions, 2], dtype=pad_samples.dtype),
[[0, pad_samples]], ops.convert_to_tensor([[0, pad_samples]]),
array_ops.zeros([num_inner_dimensions, 2], dtype=pad_samples.dtype)], array_ops.zeros([num_inner_dimensions, 2], dtype=pad_samples.dtype)
0) ], 0)
signal = array_ops.pad(signal, paddings, constant_values=pad_value) signal = array_ops.pad(signal, paddings, constant_values=pad_value)
signal_shape = array_ops.shape(signal) signal_shape = array_ops.shape(signal)