tensor pytorch conversión de tipo de datos

 Tensor tipo de datos contiene un total de datos

class DoubleTensor(Tensor): ...
class FloatTensor(Tensor): ...
class LongTensor(Tensor): ...
class IntTensor(Tensor): ...
class ShortTensor(Tensor): ...
class CharTensor(Tensor): ...
class ByteTensor(Tensor): ...
class BoolTensor(Tensor): ...

 Ejemplos:

import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.autograd import Variable
b = torch.rand(3,3)#得到的是floattensor值,
print(type(b),b,b.type())
a = b.long()#得到的是longtensor值
print(type(a),a,a.type())
d=b.int()
print(type(d),d,d.type())

El resultado:

PS F: Proyecto \ graduación \ aprendizaje profundo> pitón \ c.py.
<Clase 'torch.Tensor'> tensor ([[0.4874, 0.3303, 0.6307],
        [0.4765, 0.2642, 0.4723],
        [0,2643, 0,5587, 0,0493]] ) torch.FloatTensor
<clase 'torch.Tensor'> tensor ([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]]) torch.LongTensor
<clase 'torch.Tensor' > tensor ([[0, 0, 0],
        [0, 0, 0],
        [0, 0, 0]], dtype = torch.int32) torch.IntTensor

 

resumen:

Ver tipo de datos a.type (), el .long conversión de datos () similar al resto

Referencia: https://blog.csdn.net/hustchenze/article/details/79154139

Publicados 234 artículos originales · ganado elogios 61 · vistas 120 000 +

Supongo que te gusta

Origin blog.csdn.net/weixin_42528089/article/details/103841360
Recomendado
Clasificación