def test_RNN(self):
layer = Radial(Flipout(BayesianLSTM)(10, 10))
in_ = torch.ones(2, 3, 10)
out_, _ = layer(in_)
#print(out_)
self.assertEqual((out_[0, :, :] != out_[1, :, :]).any(),
torch.tensor(True))
def test_RNN(self):
layer = Radial(BayesianLSTM)(10, 10)
in_ = torch.ones(2, 3, 10)
out_, _ = layer(in_)
def test_linear(self):
layer = Radial(BayesianLinear)(10, 10)
in_ = torch.ones(2, 10)
out_ = layer(in_)