CANN/pypto one_hot操作文档

# pypto.one_hot

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

功能说明

将整数Tensor转换为对应的 one-hot 编码，其中每个整数被转换为一个向量，只有对应位置为1，其余为0。

函数原型

one_hot(input: Tensor, num_classes: int) -> Tensor

参数说明

返回值说明

返回一个Shape为(input, num_classes)、数据类型为DT_INT64的Tensor。

约束说明

TileShape 对输出切分，TileShape 的维度与输出一致，TileShape 的尾轴需等于 num_classes 。

调用示例

TileShape设置示例

说明：调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。

TileShape维度应和输出一致。

示例1：输入input shape为[m, n]，输出为[m, n, t], 其中t=num_classes，TileShape设置为[m1, n1, t1], 则m1, n1分别用于切分m, n轴。t1必须等于 num_classes, t轴不可切，必须保证t轴全载。

pypto.set_vec_tile_shapes(4, 16, 32)

接口调用示例

x = pypto.tensor([3], pypto.DT_INT32) y = pypto.one_hot(x, 5)

结果示例如下：

输入数据x: [0, 2, 4] 输出数据y: [[1, 0, 0, 0, 0], [0, 0, 1, 0, 0], [0, 0, 0, 0, 1]]

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto