美文网首页
pytorch 常用数据类型转换

pytorch 常用数据类型转换

作者: JeremyL | 来源:发表于2020-04-19 20:22 被阅读0次

torch.Tensor

torch.Tensor是一种包含单一数据类型元素的多维矩阵。torch.Tensor是默认的tensor类型(torch.FlaotTensor)的简称

Torch定义了七种CPU tensor类型和八种GPU tensor类型:

Data tyoe CPU tensor GPU tensor
32-bit floating point torch.FloatTensor torch.cuda.FloatTensor
64-bit floating point torch.DoubleTensor torch.cuda.DoubleTensor
16-bit floating point N/A torch.cuda.HalfTensor
8-bit integer (unsigned) torch.ByteTensor torch.cuda.ByteTensor
8-bit integer (signed) torch.CharTensor torch.cuda.CharTensor
16-bit integer (signed) torch.ShortTensor torch.cuda.ShortTensor
32-bit integer (signed) torch.IntTensor torch.cuda.IntTensor
64-bit integer (signed) torch.LongTensor torch.cuda.LongTensor

根据可选择的大小和数据新建一个tensor

如果没有提供参数,将会返回一个空的零维张量。如果提供了numpy.ndarray,torch.Tensortorch.Storage,将会返回一个有同样参数的tensor.如果提供了python序列,将会从序列的副本创建一个tensor。

  • class torch.Tensor
    class torch.Tensor(sizes)
    class torch.
    Tensor(size)
    class torch.
    Tensor(sequence or list)
    class torch.
    Tensor(ndarray)
    class torch.
    Tensor(tensor)
    class torch.
    Tensor*(storage)

Numpy array转换为Tensor

tensor = torch.from_numpy(np.array())

Tensor转换为Numpy array

np.array = tensor.numpy()

Tensor返回一个数值
torch.Tensor.item(): 是得到一个元素张量里面的元素值(只能是一个值,比如loss,acc)

Tensor返回一个列表

tolist()

将对象转为指定类型

type(new_type=None, async=False)

Variable
Variable API 几乎和 Tensor API一致 (除了一些in-place方法,这些in-place方法会修改 required_grad=Trueinput 的值)。多数情况下,将Tensor替换为Variable,代码一样会正常的工作。由于这个原因,我们不会列出Variable的所有方法,你可以通过torch.Tensor的文档来获取相关知识。

  • detach(): Returns a new Variable, detached from the current graph.

参考:
Pytorch Docs

相关文章

  • pytorch 常用数据类型转换

    torch.Tensor torch.Tensor是一种包含单一数据类型元素的多维矩阵。torch.Tensor是...

  • Python数据类型转换

    本文主要介绍Python中的常用数据类型转换,更多内容请参考:Python学习指南 常用的数据类型转换

  • 4.函数

    数据类型转换 Python内置的常用函数还包括数据类型转换函数,比如int()函数可以把其他数据类型转换为整数: ...

  • 类型的强制转换

    数据类型转换:类型名(被转换的数据) 常用数据类型:int float bool str 1.其它的数据类型转换成...

  • 2019-05-06 Python --调用函数

    数据类型转换 Python内置的常用函数还包括数据类型转换函数,比如int()函数可以把其他数据类型转换为整数: ...

  • 数据类型转换

    目标 数据类型转换的必要性 数据类型转换常用方法 一. 转换数据类型的作用 问:input()接收用户输入的数据都...

  • js常用math方法

    Math常用方法 js常用数据类型转换 toString():转换成字符串;String():吧unll、unde...

  • python 数据类型转换的函数列举和必要性

    学习Python的转换数据类型前期主要学习目标有两个,一是数据类型转换的必要性,二是数据类型转换常用方法。 一、转...

  • python 中常用数据类型转换函数有哪些?它们的使用方法是什么

    常用数据类型转换函数 在python 编程时往往我们想要的数据可能并不是理想的数据类型,这时候就需要把数据类型转换...

  • 7. 函数

    Python内置的常用函数还包括数据类型转换函数,比如int()函数可以把其他数据类型转换为整数 函数名其实就是指...

网友评论

      本文标题:pytorch 常用数据类型转换

      本文链接:https://www.haomeiwen.com/subject/jjyevhtx.html