写点什么

reshape

0 人感兴趣 · 1 次引用

  • 最新
  • 推荐
https://static001.geekbang.org/infoq/b0/b095a8229fb6aa6b5ba6f519c00e115c.jpeg?x-oss-process=image%2Fresize%2Cw_416%2Ch_234

Pytorch 基础 -tensor 数据结构

torch.Tensor 是一种包含单一数据类型元素的多维矩阵,类似于 numpy 的 array。可以使用使用 torch.tensor() 方法将 python 的 list 或序列数据转换成 Tensor 数据,生成的是dtype 默认是 torch.FloatTensor。

reshape_reshape技术文章_InfoQ写作社区