Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
tensor.py 202 Bytes
Newer Older
rusty1s's avatar
generic test  
rusty1s committed
1
2
3
4
5
6 
cpu_tensors = [
    'ByteTensor', 'CharTensor', 'ShortTensor', 'IntTensor', 'LongTensor',
    'FloatTensor', 'DoubleTensor'
]

cuda_tensors = ['cuda.{}'.format(t) for t in cpu_tensors + ['HalfTensor']]