Инициализируйте параметры как нули в 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 (последний вызов последним) в () 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

TypeError: тип данных не понятен

Answer 1

np.zeros принимает только форму в виде кортежа или одного целого числа (в случае 1-мерных массивов).Если вам нужен только одномерный массив, передайте один параметр.Если вам нужен 2d-массив, передайте его как кортеж (dim, 1).Следовательно, в зависимости от того, что вы хотите, либо используйте

w = np.zeros(dim)

, что даст вам одномерный массив нулей

, либо используйте

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

, который даст вам двумерный массив с dim числом строк и 1 столбцом.

Из официальных документов

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

Параметры:
shape: int или кортеж ints Форма нового массива, например, (2, 3) или2.