python
复制代码
import torch
b = torch.arange(0,5000).unsqueeze(-1)
'''
tensor([[ 0],
[ 1],
[ 2],
...,
[4997],
[4998],
[4999]])
'''
a = torch.arange(0,128,2)
'''
tensor([ 0, 2, 4, 6, 8, 10, 12, 14, 16, 18, 20, 22, 24, 26,
28, 30, 32, 34, 36, 38, 40, 42, 44, 46, 48, 50, 52, 54,
56, 58, 60, 62, 64, 66, 68, 70, 72, 74, 76, 78, 80, 82,
84, 86, 88, 90, 92, 94, 96, 98, 100, 102, 104, 106, 108, 110,
112, 114, 116, 118, 120, 122, 124, 126])
'''
c = b * a
'''
tensor([[ 0, 0, 0, ..., 0, 0, 0],
[ 0, 2, 4, ..., 122, 124, 126],
[ 0, 4, 8, ..., 244, 248, 252],
...,
[ 0, 9994, 19988, ..., 609634, 619628, 629622],
[ 0, 9996, 19992, ..., 609756, 619752, 629748],
[ 0, 9998, 19996, ..., 609878, 619876, 629874]])
'''