torch.meshgrid() 2022-09-02 22:34:03 浏览数 (2) x1 ,y1 = torch.meshgrid(x,y) 参数是两个,第一个参数我们假设是x,第二个参数假设就是y 输出的是两个tensor,size就是x.size * y.size(行数是x的个数,列数是y的个数) 具体输出看下面注意:两个参数的数据类型要相同,要么都是float,要么都是int,否则会报错。 int size tensor torch 数据类型 0 人点赞 上一篇:分享雷军22年前编写的代码