(x: Tensor, padding?: [number, number])
| 34 | * @return A padded 3D `tf.Tensor`. |
| 35 | */ |
| 36 | export function temporalPadding(x: Tensor, padding?: [number, number]): Tensor { |
| 37 | return tidy(() => { |
| 38 | if (x.rank !== 3) { |
| 39 | throw new ValueError( |
| 40 | `temporalPadding expects input tensor to be 3-D, but received a ` + |
| 41 | `${x.rank}-D tensor.`); |
| 42 | } |
| 43 | |
| 44 | if (padding == null) { |
| 45 | padding = [1, 1]; |
| 46 | } |
| 47 | if (padding.length !== 2) { |
| 48 | throw new ValueError( |
| 49 | `temporalPadding expects input padding pattern to be a length-2 ` + |
| 50 | `array, but received a length-${padding.length} array.`); |
| 51 | } |
| 52 | |
| 53 | const pattern: Array<[number, number]> = [[0, 0], padding, [0, 0]]; |
| 54 | return tfc.pad(x, pattern); |
| 55 | }); |
| 56 | } |
| 57 | |
| 58 | /** |
| 59 | * Pads the 2nd and 3rd dimensions of a 4D tensor. |
no test coverage detected
searching dependent graphs…