def setUp(self):
self.x = torch.rand(10, 14, 2)
self.t = torch.rand(10, 14, 2)
self.v = torch.bernoulli(torch.rand(10, 14, 1)).expand(10, 14, 2).clone()
微信
公众号
扫码关注公众号