np.zeros_like()函数

np.zeros_like()函数

w_update = np.zeros_like(x)

函数要实现构造一个和x矩阵大小一样的全零矩阵

import numpy as np
import torch

x = torch.rand(2, 3)
print(x)
w_update = np.zeros_like(x)
print(w_update)

输出：

tensor([[0.0647, 0.8316, 0.5232],
        [0.9895, 0.5264, 0.8084]])
[[0. 0. 0.]
 [0. 0. 0.]]