SpaceToDepth

SpaceToDepth - 13

版本

  • name: SpaceToDepth (GitHub)

  • domain: main

  • since_version: 13

  • function: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的算子自版本 13 起可用。

摘要

SpaceToDepth 将空间数据块重新排列到深度维度。更具体地说,此算子输出输入张量的一个副本,其中高度和宽度维度中的值被移动到深度维度。

属性

  • blocksize - INT (必需)

    移动 [blocksize, blocksize] 大小的块。

输入

  • input (异构) - T

    输入张量,形状为 [N,C,H,W],其中 N 是批量轴,C 是通道或深度,H 是高度,W 是宽度。

输出

  • output (异构) - T

    输出张量,形状为 [N, C * blocksize * blocksize, H/blocksize, W/blocksize]。

类型约束

  • T 在 ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。

SpaceToDepth - 1

版本

  • name: SpaceToDepth (GitHub)

  • domain: main

  • since_version: 1

  • function: False

  • support_level: SupportType.COMMON

  • 形状推断: True

此版本的算子自版本 1 起可用。

摘要

SpaceToDepth 将空间数据块重新排列到深度维度。更具体地说,此算子输出输入张量的一个副本,其中高度和宽度维度中的值被移动到深度维度。

属性

  • blocksize - INT (必需)

    移动 [blocksize, blocksize] 大小的块。

输入

  • input (异构) - T

    输入张量,形状为 [N,C,H,W],其中 N 是批量轴,C 是通道或深度,H 是高度,W 是宽度。

输出

  • output (异构) - T

    输出张量,形状为 [N, C * blocksize * blocksize, H/blocksize, W/blocksize]。

类型约束

  • T 在 ( tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为所有张量类型。