特牛网址导航

Pytorch学习笔记_torch查看张量shu类型-CSDN博客

网友收藏
文章浏览阅读143次。基于Anaconda下的pytorch的学习,具体安装课参考Anaconda的安装和环境使用,有详细介绍。附上pytorch官网链接:https://pytorch.org/docs/stable/index.html下面直接开始1. 查看安装pytorch的相关信息import torchimport torch.nn as nnimport torchvisionprint(torch.__version__)print(torch.version.cuda)print(torch.b_torch查看张量shu类型