pytorch-cpp icon indicating copy to clipboard operation
pytorch-cpp copied to clipboard

How to convert Type of Aten? from kByte to kFloat?

Open kirill-pinigin opened this issue 6 years ago • 1 comments

How to convert Type of Aten? from kByte to kFloat?

kirill-pinigin avatar Jun 07 '18 12:06 kirill-pinigin

@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.

warmspringwinds avatar Jun 20 '18 22:06 warmspringwinds