search

warmspringwinds / pytorch-cpp

Pytorch C++ Library
369 stars 68 forks source link

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

Open kirill-pinigin opened 6 years ago

kirill-pinigin commented 6 years ago

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

warmspringwinds commented 6 years ago

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