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

Python tf.raw_ops.DataFormatVecPermute用法及代碼示例

將輸入張量從 src_format 置換到 dst_format 。

用法

tf.raw_ops.DataFormatVecPermute(
    x, src_format='NHWC', dst_format='NCHW', name=None
)

參數

  • x 一個Tensor。必須是以下類型之一：int32 , int64。源數據格式的大小為 4 的向量或形狀 (4, 2) 的張量。
  • src_format 可選的 string 。默認為 "NHWC" 。源數據格式。
  • dst_format 可選的 string 。默認為 "NCHW" 。目標數據格式。
  • name 操作的名稱(可選)。

返回

  • 一個Tensor。具有與 x 相同的類型。

輸入張量必須是大小為 4 的向量，或 4x2 張量。

例如，使用 src_format of NHWC , dst_format of NCHW 和輸入：

[1, 2, 3, 4]

和

[[1, 2, 3, 4],
 [5, 6, 7, 8]]

，輸出將(分別)：

[1, 4, 2, 3]

和

[[1, 4, 2, 3],
 [5, 8, 6, 7]]

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.raw_ops.DataFormatVecPermute。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。