torch 学习笔记 1

rayyy9

已于 2024-08-21 16:54:20 修改

阅读量136

点赞数 3

CC 4.0 BY-SA版权

分类专栏： torch 文章标签：学习笔记深度学习 LIBTORCH

于 2024-08-21 16:54:10 首次发布

本文链接：https://blog.csdn.net/reyyy/article/details/141397854

torch 专栏收录该内容

21 篇文章

订阅专栏

#include <torch/torch.h>

torch::Tensor rand_tensor1 = torch::rand({ 4,4 });
torch::Tensor rand_tensor2 = torch::rand({ 4,4 });
torch::Tensor y = torch::matmul(rand_tensor1, rand_tensor2);
torch::Tensor A_inv = torch::linalg::inv(rand_tensor1);
torch::Tensor A_inv_b = torch::linalg::solve(rand_tensor1 , rand_tensor2 , true);