在NumPy中创建随机float和float64数组

Question

我想在NumPy中为固定的给定形状创建两个随机数组。 np.random.randn(A,B,C)创建一个随机的float64数组，如果我没记错的话，如何创建一个充满float32的数组？

如果我弄错了，那么我的问题仍然存在，反之亦然。

Answer 1

您可以在初始化后按照以下方式转换数组：

mat = np.random.randn(A,B,C)
mat = mat.astype(np.float32)

Answer 2

您可以按以下方式传递 dtype ：

np.array(np.random.randn(1,2,3), dtype=np.float32)