框架学习——pytorch
randn : 是指创建一个满足正态分布的随机张量或者矩阵
rand : 是指均匀分布的随机张量或者矩阵
x_tensor = torch.randn(2,3) 是一个尺寸为(2,3)的随机张量
y_numpy = np.random.randn(2,3) 是一个尺寸为(2,3)的随机矩阵
将张量转换为Numpy:
x_numpy = x_tensor.numpy()
将Numpy转换为张量:
y_tensor = torch.from_numpy(y_numpy)
PS:pytorch 的张量可以在GPU上进行运算,而numpy不可以