框架学习——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不可以

posted @ 2020-09-08 15:10  cfancy  阅读(122)  评论(0编辑  收藏  举报