摘要: 一个n维张量,类似于numpy,但可以在GPU上运行 搭建和训练神经网络时的自动微分/求导机制 阅读全文
posted @ 2022-02-13 20:40 xjspyx 阅读(18) 评论(0) 推荐(0) 编辑
摘要: PyTorch的tensor在概念上与numpy的array相同: tensor是一个n维数组,PyTorch提供了许多函数用于操作这些张量。任何希望使用NumPy执行的计算也可以使用PyTorch的tensor来完成,可以认为它们是科学计算的通用工具 阅读全文
posted @ 2022-02-13 20:40 xjspyx 阅读(66) 评论(0) 推荐(0) 编辑
摘要: 使用自动微分来自动计算神经网络中的后向传递。 PyTorch中的 autograd包提供了这个功能。当使用autograd时,网络前向传播将定义一个计算图;图中的节点是tensor,边是函数, 这些函数是输出tensor到输入tensor的映射。这张计算图使得在网络中反向传播时梯度的计算十分简单。 阅读全文
posted @ 2022-02-13 20:40 xjspyx 阅读(21) 评论(0) 推荐(0) 编辑