libtorch 常用API
1.常见固定值的初始化方式//zeros产生值全为0的张量 auto b = torch::zeros({ 3,4 }); //ones产生值全为1的张量 b = torch::ones({ 3,4 }); //eye产生单位矩阵张量。 b = torch::eye(4); //full产生指定值和尺寸的张量。 b = torch::full({ 3,4 }, 10); b = torch::tensor({ 33,22,11 });在这里插入代码片...
原创
2021-12-23 19:30:05 ·
894 阅读 ·
0 评论