コード例 #1
0
 def test_get_item(self):
     diag_lv = DiagLazyVariable(Variable(diag))
     diag_ev = diag_lv.evaluate()
     self.assertTrue(
         torch.equal(diag_lv[0:2].evaluate().data, diag_ev[0:2].data))
コード例 #2
0
def test_evaluate():
    diag_lv = DiagLazyVariable(Variable(diag))
    assert torch.equal(diag_lv.evaluate().data, diag.diag())
コード例 #3
0
 def test_evaluate(self):
     diag_lv = DiagLazyVariable(Variable(diag))
     self.assertTrue(torch.equal(diag_lv.evaluate().data, diag.diag()))
コード例 #4
0
def test_get_item():
    diag_lv = DiagLazyVariable(Variable(diag))
    diag_ev = diag_lv.evaluate()
    assert torch.equal(diag_lv[0:2].evaluate().data, diag_ev[0:2].data)