2022年3月11日
pytorch 常见函数
摘要： 概率分布 torch.randn_like() 函数：返回一个与 input 形状相同的服从 N(0,1) 的张量 GPU 相关 查看 torch 的版本 torch.__version__ 查看 cuda 是否 available: torch.cuda.is_available() 查看 GPU 阅读全文
posted @ 2022-03-11 14:35 guangheli 阅读(88) 评论(0) 推荐(0)