[英]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.