torch.mashgrid( )用法

原创已于 2022-03-26 18:10:51 修改 · 467 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

于 2022-03-26 17:52:27 首次发布

本文介绍如何使用PyTorch创建张量网格。通过实例演示了如何利用meshgrid函数从一维张量创建二维网格，详细解释了每个输出张量的填充方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

官方：取N个张量，每个张量可以是标量，也可以是一维向量，然后创建N个N维的网格，其中第i个网格是通过将第i个输入扩展到其他输入定义的维度来定义的。

例子：

import torch
# 1x4和1x3的tensor
x = torch.tensor([1, 2, 3, 4, 5, 6])
y = torch.tensor([7, 8, 9])
rx, ry = torch.meshgrid(x, y)
print(x)
print(y)
print(rx)
print(ry)

结果：

tensor([1, 2, 3, 4, 5, 6])
tensor([7, 8, 9])
tensor([[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3],
        [4, 4, 4],
        [5, 5, 5],
        [6, 6, 6]])
tensor([[7, 8, 9],
        [7, 8, 9],
        [7, 8, 9],
        [7, 8, 9],
        [7, 8, 9],
        [7, 8, 9]])

输入1×6和1×3的一维张量，输出两个二维张量。二维张量为6×3。第一个输出按行方向依照第一个输入的内容填充，并按列方向复制填充满。第二个输出按列方向依照第二个输入的内容填充，并按行方向复制填充满。