今天读代码遇到了没见过的函数。
pred = pred.t()
自己尝试后发现是对tensor进行转置操作
import torch
a = torch.tensor([[1,2],[3,4]])
a
a.t()
运行结果:
tensor([[1, 2],
[3, 4]])
tensor([[1, 3],
[2, 4]])
今天读代码遇到了没见过的函数。
pred = pred.t()
自己尝试后发现是对tensor进行转置操作
import torch
a = torch.tensor([[1,2],[3,4]])
a
a.t()
运行结果:
tensor([[1, 2],
[3, 4]])
tensor([[1, 3],
[2, 4]])