DEEP LEARNING WITH PYTORCH: A 60 MINUTE BLITZ | TENSORS

TENSORS

  张量是一种特殊的数据结构,非常类似于数组合矩阵。在PyTorch中,我们使用tensor 编码模型的输入和输出,以及模型的参数。
  Tensor类似于NumPy的ndarrays,除了tensor可以在GPUs或其它特殊的硬件上运行以加速运算。如果你对ndarrays非常熟悉,那么你对Tensor API也会很熟悉。如果不是,按照此快速API进行操作。

import torch
import numpy as np

import

上一篇:PyTorch与张量简介


下一篇:NumPy和Tensor之间得转换