
libtorch/C++
Lambda_eta
这个作者很懒,什么都没留下…
展开
-
[libtorch/C++]clone与copy
int main() { auto x = torch::zeros({2, 2}); auto y = x; auto m = x.clone(); auto n = torch::zeros_like(x); n.copy_(x); x[0][0] = 1; cout << y[0][0] << m[0][0] << n[0][0]; return 0; } 输出1,0,0 说明直接赋值是共享内存的。 根据文档,clone会被记录到计原创 2020-09-07 20:00:15 · 1043 阅读 · 0 评论 -
[libtorch/C++] reflection_pad2d
在pytorch1.6中,必须如下调用 x =torch::reflection_pad2d(x,{1,1,1,1});原创 2020-09-07 19:51:44 · 382 阅读 · 0 评论