繁体   English   中英

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

[英]what is the equivalent keras function for from_numpy in torch?

我在火炬中找到了一个代码,必须将其更改为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东西? 我感谢您的帮助。

您可以使用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')

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

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM