megengine.functional.expand_dims

expand_dims(inp, axis)[源代码]

在给定的axis前添加维度。

参数
  • inp – 输入张量。

  • axis – 若干新轴的位置

返回

输出张量。

例如:

import numpy as np
from megengine import tensor
import megengine.functional as F

x = tensor([1, 2])
out = F.expand_dims(x, 0)
print(out.numpy().shape)

输出:

(1, 2)