эквивалент TF.cast в Pytorch?

Question

Я новичок в PyTorch. TensorFlow имеет API tf.cast () и tf.shape () . tf.cast имеет специальное назначение в TensorFlow, есть ли что-нибудь эквивалентное в факеле? у меня есть тензор x = тензор (форма (128,64,32,32)): tf.shape (x) создать тензор измерения 1 x.shape создать истинное измерение. мне нужно использовать tf.shape (x) в факеле.

tf.cast играет другую роль, чем просто смена тензора dtype в факеле.

Кто-нибудь имел эквивалентный API в torch / PyTorch.

Answer 1

Ознакомьтесь с документацией PyTorch

Как они упоминали:

print(x.dtype) # Prints "torch.int64", currently 64-bit integer type
x = x.type(torch.FloatTensor)
print(x.dtype) # Prints "torch.float32", now 32-bit float
print(x.float()) # Still "torch.float32"
print(x.type(torch.DoubleTensor)) # Prints "tensor([0., 1., 2., 3.], dtype=torch.float64)"
print(x.type(torch.LongTensor)) # Cast back to int-64, prints "tensor([0, 1, 2, 3])"