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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。