Tf. Cast Equivalent in Pytorch?
I Am New to Pytorch. Tensorflow Has an Api Tf. Cast() and Tf. Shape(). the Tf. Cast Has Specific Purpose in Tensorflow, Is There Anything Equivalent in Torch?...
I am new to PyTorch. TensorFlow has an API tf.cast() and tf.shape(). the tf.cast has specific purpose in TensorFlow, is there anything equivalent in torch? i have tensor x= tensor(shape(128,64,32,32)): tf.shape(x) create tensor of dimension 1 x.shape create the true dimension. i need to use tf.shape(x) in torch.
tf.cast has a different role than just changing tensor dtype in torch.
did anyone have equivalent API in torch/PyTorch.
1 Answer
Check out the PyTorch Documentation
As they mentioned:
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])"
Chloe Bennett explores the intersection of pop culture, streaming entertainment, digital trends, and contemporary lifestyle. Her weekly commentary reaches thousands of culture enthusiasts.
