Hello,
I try to convert an existing code in pytorch to tensorflow. In the original code, they use “dtype=torch.cuda.LongTensor” to specify using GPU. Is there any alternative in tensorflow ? I have tried to use classical types of tensorflow such as “dtype =tf.dtypes.int64” but the code is very slow.
Thank you for your help.