PyTorch深度学习总结
第五章 PyTorch中张量(Tensor)计算操作
文章目录
前言
上文介绍了PyTorch中张量(Tensor)的拆分和拼接操作，本文将介绍张量的计算操作。
一、张量比较大小
1、torch.allclose()
2、torch.eq()和torch.equal()
3、ge、gt、le、lt、ne函数
4、torch.isnan()
二、基本运算
1、四则运算(加减乘除)
加减乘除运算为：+、-、*、/
整除：//
幂运算为：torch.pow()或**