PyTorch基本数据类型(一)
PyTorch基础入门一:PyTorch基本数据类型Pytorch里面处理的最基本的操作对象就是Tensor(张量),它表示的其实就是一个多维矩阵,并有矩阵相关的运算操作。在使用上和numpy是对应的,它和numpy唯一的不同就是,pytorch可以在GPU上运行,而numpy不可以。当然,二者也可以相互转换。Tensor的基本数据类型有五种:。其实定义的方式和numpy一样,直接传入相应的矩阵即可即可。Tensor只需要调用cuda()函数就可以将其转化为能在GPU上运行的类型。我们可以通过torch.cuda.is_available()函数来判断当前的环境是否支持GPU,如果支持,则返回True。代值检验可得计算结果是正确的。
用户评论