2021年10月10日
permute与transpose
摘要： torch.transpose(input, dim0, dim1) tensor.transpose(dim0, dim1) tensor.permute(dim0, dim1, ```,dimk) 都是交换维度的函数 transpose每次只能换两个维度，两种写法，参数顺序无所谓 permute 阅读全文
posted @ 2021-10-10 10:26 WTSRUVF 阅读(261) 评论(0) 推荐(0)