从零学习PyTorch(2)----生活总会让人晕头转向

数据介绍

numpy数据格式 “array” 可以大体上理解为 多维数组(和列表格式有点像的)，但并不是列表按照列表方式声明就是numpy，它必须按照自己的声明方式才行。

torch数据格式 “tensor” 可以大体上理解为 向量，可以将很多区域的数据转化为torch数据。

数据转换

这里留下3个示例进行数据转换

示例1 numpy2torch/torch2numpy

import numpy as np
import torch as tf

np_data = np.arange(6).reshape((2, 3))
torch_data = tf.from_numpy(np_data)
tensor2array = torch_data.numpy()

print(
    '\nnumpy', np_data,
    '\ntorch', torch_data,
    '\ntensor2array', tensor2array,
)

示例2 list2torch&abs

data = [-1, -2, 1, 2]
tensor = tf.FloatTensor(data)

print(
    '\nabs', 
    '\nnumpy', np.abs(data),
    '\ntorch', tf.abs(tensor),
)

示例3 list2numpy&矩阵相乘

import numpy as np
import torch as tf

data = [[1, 2], [3, 4]]
data = np.array(data)
tensor = tf.FloatTensor(data)

print(
    '\nnumpy', data.dot(data),
    # '\ntorch', tensor.dot(tensor)
)