NumPy np.ones函数的具体使用

numpy.ones(shape, dtype=none, order='c', *, like=none)

import numpy as np

# 1. 创建1维数组
arr1d = np.ones(5)  # array([1., 1., 1., 1., 1.])

# 2. 创建2维矩阵
arr2d = np.ones((2, 3), dtype=int)  # array([[1, 1, 1], [1, 1, 1]])

# 3. 创建3维张量
arr3d = np.ones((2, 2, 2), dtype=np.float32)

input_size = 100
hidden_size = 50
weights = np.ones((input_size, hidden_size)) * 0.01  # 小值初始化

data = np.array([1, 3, 5, 7, 9])
mask = np.ones(5, dtype=bool)
mask[[1, 3]] = false  # array([ true, false,  true, false,  true])
filtered = data[mask]  # array([1, 5, 9])

# 创建单位矩阵（等价于np.eye(3)）
identity = np.ones((3, 3)) * np.eye(3)

# 生成指定范围的等差数列
start, 终止 = 2, 10
steps = np.ones(5) * np.linspace(start, 终止, 5)