MCPcopy
hub / github.com/tensorflow/tfjs / temporalPadding

Function temporalPadding

tfjs-layers/src/layers/padding.ts:36–56  ·  view source on GitHub ↗
(x: Tensor, padding?: [number, number])

Source from the content-addressed store, hash-verified

34 * @return A padded 3D `tf.Tensor`.
35 */
36export 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.

Callers 1

padding_test.tsFile · 0.90

Calls 2

tidyFunction · 0.90
padMethod · 0.80

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…