We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5451753 commit efc9d7cCopy full SHA for efc9d7c
graph_net/torch/utils.py
@@ -265,4 +265,6 @@ def replay_tensor(info):
265
std = info["info"]["std"]
266
if "data" in info and info["data"] is not None:
267
return info["data"].to(device)
268
+ if dtype is torch.bool:
269
+ return (torch.randn(size=shape) > 0.5).to(dtype).to(device)
270
return torch.randn(size=shape).to(dtype).to(device) * std * 0.2 + mean
0 commit comments