當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


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。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。