I am converting a keras code to pytorch and want to convert this
import keras.backend as K
train_y = np.array(train_y, dtype=K.floatx())
I am trying to find a way to include the CUDA float variable but couldn’t. Any help is appreciated