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
相关文章
相关标签/搜索