pytorch學習筆記

一 pytorch數據結構 Tensor有自己的數據類型，也有CPU和GPU版本的Tensor。torch.set_default_tensor_type修改指定tensor類型，默認是torch.float32 or torch.float。 每個元素佔用32bit/8=4Byte內存。 torch和numpy 二者用法類似，二者可以互相轉換。 有共享內存和不共享內存的情況 1.numpy轉te

>>阅读原文<<