Numpy

官方文档：https://www.numpy.org.cn/article/

numpy中的数组

import numpy as np

my_array = np.array([1,2,3,4,5])
print(my_array,my_array.shape,my_array[0])
my_array[0] = -2
print(my_array)
my_new_array = np.zeros((5))
print(my_new_array)
my_random_array = np.random.random((5))
print(my_random_array)
my_2d_array = np.zeros((2,3))
print(my_2d_array)
my_2d_array_new = np.ones((2, 4))
print(my_2d_array_new)
my_array = np.array([[4,5,4],[6,1,3]])
print(my_array[0][1],my_array.shape)
print(my_array[1,:],my_array[:,1])

numpy与torch的互换

import torch
a = torch.ones([5,3])
b = a.numpy()
c = torch.from_numpy(b)
print(a,b,c)