对pytorch的张量使用for循环

1. 一维张量

对于一个一维张量，使用for循环，取出的是里面的每一项

import torch
a = torch.tensor[1, 2, 3]
for b in a:
	print(b)

此时输出结果为

tensor(1)
tensor(2)
tensor(3)

2. 二维张量

代码

import torch
a = torch.tensor([[1, 2, 3], [4, 5, 6]])
for b in a:
	print(b)

输出结果

tensor([1, 2, 3])
tensor([4, 5, 6])

因为是二维向量，可以再使用一次for循环，修改代码如下

import torch
a = torch.tensor([1, 2, 3], [4, 5, 6])
for b in a:
	for c in b:
		print(c)

此时输出结果为

tensor(1)
tensor(2)
tensor(3)
tensor(4)
tensor(5)
tensor(6)

3. 总结

一个for循环相当于剥掉了一个外壳，也就是这里的[ ]（中括号），最终输出是也是张量，维度减一