随笔分类 - LibTorch
PyTorch C++ API
摘要:0. 写在前面 本文将使用基于LibTorch(PyTorch C++接口)的神经网络求解器,对一维稳态对流扩散方程进行求解,文中仅对神经网络求解器对特定问题的求解能力进行了介绍,单纯一个无监督学习问题,未涉及迁移到其他问题的适用性问题等。水平有限,如有问题还希望读者斧正。研究问题参考自教科书$^{
阅读全文
摘要:最近在学习过程中需要对服从某种分布的离散型随机变量进行抽样,在LibTroch中查到了torch::multinomial(多项分布),该方法的接口如下: at::Tensor multinomial( const at::Tensor & self, // 概率分布 int64_t num_sam
阅读全文

浙公网安备 33010602011771号