Pytorch 在Tensor和numpy之间相互转换

Song • 592 次浏览 • 0个评论 • 2018年07月11日

在Pytorch 中Tensor和numpy之间可以相互转换，Tensor转换为numpy很简单，只需要numpy()函数即可，Numpy转化为Tensor也很简单，只需torch.from_numpy(b)即可： [attach]25[/attach][code]# -- coding: utf-8 -- import numpy as np import torch import torch.nn as nn import torch.nn.functional as F

a = torch.Tensor(3,3) print a b = a.numpy() print torch.from_numpy(b)[/code]

提交评论

请登录后评论

用户评论

    当前暂无评价,快来发表您的观点吧...

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包，是Python优先的深度学习框架。作为numpy的替代品；使用强大的GPU能力，提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。