Open kirill-pinigin opened 6 years ago
How to convert Type of Aten? from kByte to kFloat?
@kirill-pinigin Similar to this line: https://github.com/warmspringwinds/pytorch-cpp/blob/master/examples/read_allocated_gpu_memory.cpp#L65
Just replace with your desired type.
How to convert Type of Aten? from kByte to kFloat?