Python Variable._dimV 예제들

예제 #1

파일: test_sparse.py 프로젝트: Jsmilemsj/pytorch

 def test_cuda_empty(self):
     from torch.autograd import Variable
     x = Variable(torch.sparse.FloatTensor(2, 3, 4))
     y = x.cuda(0)
     self.assertEqual(x._dimI(), y._dimI())
     self.assertEqual(x._dimV(), y._dimV())
     x = y.cpu()
     self.assertEqual(y._dimI(), x._dimI())
     self.assertEqual(y._dimV(), x._dimV())

예제 #2

파일: test_sparse.py 프로젝트: steveatgit/pytorch

 def test_cuda_empty(self):
     from torch.autograd import Variable
     x = Variable(torch.sparse.FloatTensor(2, 3, 4))
     y = x.cuda(0)
     self.assertEqual(x._dimI(), y._dimI())
     self.assertEqual(x._dimV(), y._dimV())
     x = y.cpu()
     self.assertEqual(y._dimI(), x._dimI())
     self.assertEqual(y._dimV(), x._dimV())