pytorch-cpp
pytorch-cpp copied to clipboard
How to convert Type of Aten? from kByte to kFloat?
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.