tf.signal: Remove set_shape from documentation for linear_to_mel_weight_matrix.

tf.tensordot has shape inference now.

PiperOrigin-RevId: 311042725
Change-Id: I4ded418173bdcc5238d05f73a12ca8e387bbac04
This commit is contained in:
RJ Skerry-Ryan 2020-05-11 19:19:37 -07:00 committed by TensorFlower Gardener
parent 2483676ae7
commit cf4fb9ed5e

View File

@ -128,8 +128,6 @@ def linear_to_mel_weight_matrix(num_mel_bins=20,
# S has shape [..., num_spectrogram_bins].
# M has shape [..., num_mel_bins].
M = tf.tensordot(S, A, 1)
# tf.tensordot does not support shape inference for this case yet.
M.set_shape(S.shape[:-1].concatenate(A.shape[-1:]))
Args:
num_mel_bins: Python int. How many bands in the resulting mel spectrum.