将tensor按维度分块,返回一个元组
例
import torch
a = torch.Tensor([[1,2,2],[4,5,6],[3,5,8], [4,6,7]])
b = torch.chunk(a,2,dim=0)
print(b)
将tensor按维度分块,返回一个元组
例
import torch
a = torch.Tensor([[1,2,2],[4,5,6],[3,5,8], [4,6,7]])
b = torch.chunk(a,2,dim=0)
print(b)