Unsqueeze

Unsqueeze - 24

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • 起始版本：24

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符已可用于版本 24 及以上。

摘要

为输入张量（data）的形状插入单一维度的条目。它需要一个必需的输入axes——其中包含一个维度索引列表，该算子将在输出张量（expanded）的相应索引处插入一个值为1的维度。

例如，给定一个形状为 [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中值的顺序无关紧要，可以是任何顺序。

输入

  • data (异构) - T

    原始张量

  • axes (异构) - tensor(int64)

    表示要插入的维度的 1D 整型张量。负值表示从后往前计算维度。可接受的范围是 [-r, r-1]，其中 r = rank(expanded)。

输出

  • expanded (异构) - T

    重塑后的张量，数据与输入相同。

类型约束

  • T 在 ( 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(float8e8m0), tensor(int16), tensor(int32), tensor(int4), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint4), tensor(uint64), tensor(uint8) )

    将输入和输出类型限制为 IRv12 及以前的所有张量类型。

Unsqueeze - 23

版本

  • 名称: Unsqueeze (GitHub)

  • 域: main

  • 起始版本：23

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符已可用于版本 23 及以上。

摘要

为输入张量（data）的形状插入单一维度的条目。它需要一个必需的输入axes——其中包含一个维度索引列表，该算子将在输出张量（expanded）的相应索引处插入一个值为1的维度。

例如，给定一个形状为 [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中值的顺序无关紧要，可以是任何顺序。

输入

  • data (异构) - T

    原始张量

  • axes (异构) - tensor(int64)

    表示要插入的维度的 1D 整型张量。负值表示从后往前计算维度。可接受的范围是 [-r, r-1]，其中 r = rank(expanded)。

输出

  • expanded (异构) - T

    重塑后的张量，数据与输入相同。

类型约束

  • T 在 ( 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

  • since_version: 21

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的操作符自 版本 21 起可用。

摘要

为输入张量（data）的形状插入单一维度的条目。它需要一个必需的输入axes——其中包含一个维度索引列表，该算子将在输出张量（expanded）的相应索引处插入一个值为1的维度。

例如，给定一个形状为 [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中值的顺序无关紧要，可以是任何顺序。

输入

  • 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(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]) 输出一个张量（expanded），该张量包含与data相同的数据，但形状为 [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

  • 起始版本: 11

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

此版本的运算符自 版本 11 起可用。

总结

为输入张量（data）的形状插入单一维度的条目。它需要一个必需的参数axes——其中包含一个维度索引列表，该算子将在输出张量（expanded）的相应索引处插入一个值为1的维度。

例如：给定一个形状为 [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

  • 起始版本: 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) )

    将输入和输出类型限制为所有张量类型。