pad¶

dragon.vm.tensorflow.pad(
  tensor,
  paddings,
  mode='CONSTANT',
  constant_values=0,
  name=None
)[source]¶

Pad the input according to the given sizes.

paddings should be a sequence with \(N\) tuples, where \(N\) is the number of input dimensions:

x = tf.ones(shape=(2, 3))
print(tf.pad(x, [[0, 1], [1, 0]]))  # Ok, (2, 3) -> (3, 4)
print(tf.pad(x, [[0, 1]]))  # Wrong

Following padding modes are supported:

x = tf.constant([[1, 2, 3], [4, 5, 6]])

# ConstantPad
print(tf.pad(x, [[0, 1], [1, 0]], 'CONSTANT', 9))

# ReflectPad
print(tf.pad(x, [[0, 1], [1, 0]], 'REFLECT'))

Parameters:

• tensor (dragon.Tensor) – The input tensor.
• paddings (Sequence[Tuple[int]]) – The begins and ends of padding.
• mode ({'CONSTANT', 'REFLECT'}, optional) – The padding mode.
• constant_values (int, optional, default=0) – The constant value in CONSTANT mode.
• name (str, optional) – The operation name.

Returns:

dragon.Tensor – The output tensor.