megengine.functional.nn.indexing_one_hot

indexing_one_hot(src, index, axis=1, keepdims=False)

对一些轴进行One-hot索引。

参数

• src (Tensor) – 输入张量。

• index (Tensor) – 索引张量。

• axis (int) – 源数据上的轴，索引值为其索引。 默认： 1

• keepdims – 是否在结果数据中删除该轴。 默认： False

实际案例

>>> src = Tensor([[1.0, 2.0]])
>>> index = Tensor([0])
>>> val = F.indexing_one_hot(src, index)
>>> val.numpy()
array([1.], dtype=float32)

返回类型

Tensor

上一页

megengine.functional.nn.one_hot

下一页

megengine.functional.nn.embedding