我想知道如何使链接器支持float64精度计算。以下是我在链接器中编写的简单的自动编码器代码。最初我是用float32 dtype运行它的,但是现在我想测试一下float64的精度,我该怎么做?
class AutoEncoder(chainer.Chain):
def __init__(self, num_input=1024, num_hidden=512, activate=F.leaky_relu, dropout_ratio=0.0):
super(AutoEncoder, self).__init__(
encoder=L.Linear(num_input, num_hidden),
decoder=L.Linear(num_hidden, num_input))
self.num_input = num_input
self.num_hidden = num_hidden
self.activate = activate
self.dropout_ratio = dropout_ratio
def __call__(self, x, hidden=False):
h = F.dropout(self.activate(self.encoder(x)), ratio=self.dropout_ratio)
if hidden:
return h
y = F.dropout((self.decoder(h)), ratio=self.dropout_ratio)
return y
当我尝试使用float64数据运行它时,会引发dtype不匹配错误。