我们可能会经常在阅读深度学习代码时候看到:
tensor = torch.as_tensor(numpy数组)
为什么要把numpy转为tensor呢?因为tensor是专门为GPU加速设计的矩阵,而numpy却不行。其实就相当于一个为GPU设计的数据结构。
我们可能会经常在阅读深度学习代码时候看到:
tensor = torch.as_tensor(numpy数组)
为什么要把numpy转为tensor呢?因为tensor是专门为GPU加速设计的矩阵,而numpy却不行。其实就相当于一个为GPU设计的数据结构。