meshgrid

    该OP的输入是张量或者包含张量的列表, 包含 k 个一维张量,对每个张量做扩充操作,输出 k 个 k 维张量。

    返回: k 个 k 维张量,每个张量的形状均为(N1, N2, …, Nk)。

    代码示例