Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
tensor.py 201 Bytes
Newer Older
rusty1s's avatar
generic test  
rusty1s committed
1
2
3
4
5 
cpu_tensors = [
    'ByteTensor', 'CharTensor', 'ShortTensor', 'IntTensor', 'LongTensor',
    'FloatTensor', 'DoubleTensor'
]
rusty1s's avatar
clean up  
rusty1s committed
6 
gpu_tensors = ['cuda.{}'.format(t) for t in cpu_tensors + ['HalfTensor']]