在 Python 中将参数初始化为零

Question

我尝试实现参数初始化并收到错误消息：

import numpy as np

def initialize_with_zeros(dim):
    w = np.zeros(dim, 1)
    b = 0
    return w, b

dim = 2
initialize_with_zeros(dim)

---

这是错误：

TypeError Traceback（最近一次调用最后一次） in () 5 6 dim = 2 ----> 7 initialize_with_zeros(dim)

在 initialize_with_zeros(dim) 1 def initialize_with_zeros(dim): ----> 2 w = np.zeros(dim, 1) 3 b = 0 4 return w, b 5

类型错误：无法理解数据类型

Answer 1

np.zeros仅将形状作为元组或单个整数（在一维数组的情况下）。 如果您只需要一个一维数组，请传递一个参数。 如果您需要二维数组，请作为元组 (dim,1) 传递。 因此，取决于你想要什么，要么使用

w = np.zeros(dim)

这会给你一个一维的零数组

或使用

w = np.zeros((dim, 1))

这将给你与二维阵列dim的行数和1列。

来自官方文档

numpy.zeros(shape, dtype=float, order='C')

参数：
shape : int 或 int 元组新数组的形状，例如 (2, 3) 或 2。

Answer 2

用零初始化参数：

# GRADED FUNCTION: initialize_with_zeros
import numpy as np

def initialize_with_zeros(dim):
    w = np.zeros([dim, 1])
    b = 0
    return w, b

dim = 2
w,b=initialize_with_zeros(dim)

print ("w = " + str(w))
print ("b = " + str(b))