oneflow
oneflow copied to clipboard
nn.functional.layer_norm输出与pytorch不一致
Description
以下代码,在传入参数相同的情况下,oneflow和pytorch输出结果不一致 import oneflow as torch with torch.autocast(enabled=False, device_type=module_device.type): return torch.nn.functional.layer_norm(downcast_x, self.normalized_shape, downcast_weight, downcast_bias, self.eps)