当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python tf.raw_ops.PadV2用法及代码示例


填充张量。

用法

tf.raw_ops.PadV2(
    input, paddings, constant_values, name=None
)

参数

  • input 一个Tensor
  • paddings 一个Tensor。必须是以下类型之一:int32 , int64
  • constant_values 一个Tensor。必须与 input 具有相同的类型。
  • name 操作的名称(可选)。

返回

  • 一个Tensor。具有与 input 相同的类型。

此操作根据您指定的paddingsconstant_values 填充inputpaddings 是形状为 [Dn, 2] 的整数张量,其中 n 是 input 的秩。对于每个维度,input , paddings[D, 0]的D表示该维度中input的内容之前要添加多少个填充值,paddings[D, 1]表示该维度中input的内容之后要添加多少个填充值。 constant_values 是与 input 类型相同的标量张量,指示用于填充 input 的值。

输出的每个维度 D 的填充大小为:

paddings(D, 0) + input.dim_size(D) + paddings(D, 1)

例如:

# 't' is [[1, 1], [2, 2]]
# 'paddings' is [[1, 1], [2, 2]]
# 'constant_values' is 0
# rank of 't' is 2
pad(t, paddings) ==> [[0, 0, 0, 0, 0, 0]
                      [0, 0, 1, 1, 0, 0]
                      [0, 0, 2, 2, 0, 0]
                      [0, 0, 0, 0, 0, 0]]

相关用法


注:本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.raw_ops.PadV2。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。