Unsqueeze

Unsqueeze - 23

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • 自版本: 23

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符已自版本 23开始可用。

摘要

将单维度条目插入输入张量的形状 (data)。它接受一个必需的输入 axes，其中包含维度索引列表，并且此运算符将在输出张量 (expanded) 的对应索引中插入值为 1 的维度。

例如，给定形状为 [3, 4, 5] 的输入张量 (data)，则 Unsqueeze(data, axes=[0, 4]) 输出一个包含与 data 相同数据的张量 (expanded)，但形状为 [1, 3, 4, 5, 1]。

输入 axes 不应包含任何重复条目。如果包含重复条目，则为错误。输出张量的秩 (output_rank) 是输入张量的秩 (data) 加上 axes 中的值数量。 axes 中的每个值都应在 (包含) 范围 [-output_rank , output_rank - 1] 内。 axes 中值的顺序无关紧要，可以按任何顺序出现。

输入

  • data (异构) - T

    原始张量

  • axes (异构) - tensor(int64)

    表示要插入的维度的整数列表。负值表示从后面计算维度。接受的范围是 [-r, r-1]，其中 r = rank(expanded)。

输出

  • expanded (异构) - T

    与输入具有相同数据的重塑张量。

类型约束

  • T in ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(float4e2m1), tensor(float8e4m3fn), tensor(float8e4m3fnuz), tensor(float8e5m2), tensor(float8e5m2fnuz), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为 IRv11 之前的所有张量类型。

Unsqueeze - 21

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • 自版本: 21

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符已自版本 21开始可用。

摘要

将单维度条目插入输入张量的形状 (data)。它接受一个必需的输入 axes，其中包含维度索引列表，并且此运算符将在输出张量 (expanded) 的对应索引中插入值为 1 的维度。

例如，给定形状为 [3, 4, 5] 的输入张量 (data)，则 Unsqueeze(data, axes=[0, 4]) 输出一个包含与 data 相同数据的张量 (expanded)，但形状为 [1, 3, 4, 5, 1]。

输入 axes 不应包含任何重复条目。如果包含重复条目，则为错误。输出张量的秩 (output_rank) 是输入张量的秩 (data) 加上 axes 中的值数量。 axes 中的每个值都应在 (包含) 范围 [-output_rank , output_rank - 1] 内。 axes 中值的顺序无关紧要，可以按任何顺序出现。

输入

  • data (异构) - T

    原始张量

  • axes (异构) - tensor(int64)

    表示要插入的维度的整数列表。负值表示从后面计算维度。接受的范围是 [-r, r-1]，其中 r = rank(expanded)。

输出

  • expanded (异构) - T

    与输入具有相同数据的重塑张量。

类型约束

  • T in ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(float8e4m3fn), tensor(float8e4m3fnuz), tensor(float8e5m2), tensor(float8e5m2fnuz), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) )

    将输入和输出类型约束为 IRv10 之前的所有张量类型。

Unsqueeze - 13

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • 自版本: 13

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符已自版本 13开始可用。

摘要

将单维度条目插入输入张量的形状 (data)。它接受一个必需的输入 axes，其中包含维度索引列表，并且此运算符将在输出张量 (expanded) 的对应索引中插入值为 1 的维度。

例如，给定形状为 [3, 4, 5] 的输入张量 (data)，则 Unsqueeze(data, axes=[0, 4]) 输出一个包含与 data 相同数据的张量 (expanded)，但形状为 [1, 3, 4, 5, 1]。

输入 axes 不应包含任何重复条目。如果包含重复条目，则为错误。输出张量的秩 (output_rank) 是输入张量的秩 (data) 加上 axes 中的值数量。 axes 中的每个值都应在 (包含) 范围 [-output_rank , output_rank - 1] 内。 axes 中值的顺序无关紧要，可以按任何顺序出现。

输入

  • data (异构) - T

    原始张量

  • axes (异构) - tensor(int64)

    表示要插入的维度的整数列表。负值表示从后面计算维度。接受的范围是 [-r, r-1]，其中 r = rank(expanded)。

输出

  • expanded (异构) - T

    与输入具有相同数据的重塑张量。

类型约束

  • 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) )

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

Unsqueeze - 11

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • since_version: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符已从版本 11 开始提供。

概述

将单维条目插入输入张量 (data) 的形状中。接受一个必需参数 axes - 它包含一个维度索引列表，该运算符将插入一个值为 1 的维度到输出张量 (expanded) 的相应索引中。

例如：给定一个形状为 [3, 4, 5] 的输入张量 (data)，则 Unsqueeze(data, axes=[0, 4]) 输出一个张量 (expanded)，它包含与 data 相同的数据，但形状为 [1, 3, 4, 5, 1]。

属性 axes 不应包含任何重复条目。如果包含重复条目，则会发生错误。输出张量 (output_rank) 的秩等于输入张量 (data) 的秩加上 axes 中值的个数。 axes 中的每个值都应在 (包含) 范围 [-output_rank , output_rank - 1] 内。 axes 中值的顺序无关紧要，可以以任何顺序出现。

属性

  • axes - INTS (必需)

    表示要插入的维度的整数列表。负值表示从后面计算维度。接受的范围是 [-r, r-1]，其中 r = rank(expanded)。

输入

  • data (异构) - T

    原始张量

输出

  • expanded (异构) - T

    与输入具有相同数据的重塑张量。

类型约束

  • 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) )

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

Unsqueeze - 1

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • since_version: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符已从版本 1 开始提供。

概述

将单维条目插入张量的形状中。接受一个必需参数 axes，这是一个将被插入的维度的列表。 axes 中的维度索引与输出张量中看到的维度索引相同。例如：给定一个形状为 [3, 4, 5] 的张量，则 Unsqueeze(tensor, axes=[0, 4]) 的形状为 [1, 3, 4, 5, 1]

属性

  • axes - INTS (必需)

    非负整数列表，表示要插入的维度。

输入

  • data (异构) - T

    原始张量

输出

  • expanded (异构) - T

    与输入具有相同数据的重塑张量。

类型约束

  • 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) )

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