我对PyTorch还不熟悉。TensorFlow有API tf.cast()和tf.shape()。在TensorFlow中,tf.cast具有特定的用途,那么在torch中是否有等效的功能? 我有一个张量x=tensor(shape(128,64,32,32)):tf.shape(x)创建1维张量,而x.shape创建真实的维度。我需要在torch中使用tf.shape(x)。
tf.cast在torch中的作用与仅改变张量dtype不同。
是否有人知道在torch/PyTorch中是否有相应的API?