火炬中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')

除此之外，您使用的代碼是正確的。