x1 ,y1 = torch.meshgrid(x,y)
参数是两个，第一个参数我们假设是x，第二个参数假设就是y
输出的是两个tensor，size就是x.size * y.size（行数是x的个数，列数是y的个数）
具体输出看下面：
注意：两个参数的数据类型要相同，要么都是float，要么都是int，否则会报错。
torch.meshgrid()
猜你喜欢
转载自blog.csdn.net/weixin_36670529/article/details/105350139
今日推荐
周排行