火炬中from_numpy的等效keras函数是什么？

Question

我在火炬中找到了一个代码，必须将其更改为keras，但是我找不到其中的一些等效代码。例如，我将其中一些更改如下，但是我不确定它们是否正确：

 `torch.tensor` to `K.variable` ( `K` is `from keras import backend as K`)
  unsqueez_(1) to K.expand_dims
  torch.empty((3,) + requested_shape) to K.zeros((3,) + requested_shape)

但是我找不到torch.from_numpy的任何东西。现在，我的问题是关于我所做的上述更改，它们是真的吗？还有类似于torch.from_numpy的东西吗？感谢您的帮助。

Answer 1

您可以使用numpy数组初始化变量，如下所示：

ary = np.random.normal(size=(2, 2))
v = K.variable(ary)

或使用cast()将numpy数组转换为张量：

ary = np.random.normal(size=(2, 2))
tensor = K.cast(ary, dtype='float32')

除此之外，您使用的代码是正确的。