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

Merge pull request 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

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)
# Pad the inner dimension of signal by pad_samples.
paddings = array_ops.concat(
[array_ops.zeros([num_outer_dimensions, 2], dtype=pad_samples.dtype),
[[0, pad_samples]],
array_ops.zeros([num_inner_dimensions, 2], dtype=pad_samples.dtype)],
0)
paddings = array_ops.concat([
array_ops.zeros([num_outer_dimensions, 2], dtype=pad_samples.dtype),
ops.convert_to_tensor([[0, pad_samples]]),
array_ops.zeros([num_inner_dimensions, 2], dtype=pad_samples.dtype)
], 0)
signal = array_ops.pad(signal, paddings, constant_values=pad_value)
signal_shape = array_ops.shape(signal)