写点什么

ndarray

0 人感兴趣 · 2 次引用

  • 最新
  • 推荐
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。

ndarray 的创建及属性

用户头像
向阳逐梦
2022-10-30

上节课我们有提到过 ndarray 的创建,我只需要将列表传入到 np.array() 函数中。例如:

ndarray_ndarray技术文章_InfoQ写作社区