PyTorch中Tensor的维度变换
摘要:
对于 PyTorch 的基本数据对象 Tensor (张量),在处理问题时,需要经常改变数据的维度,以便于后期的计算和进一步处理,本文旨在列举一些维度变换的方法并举例,方便大家查看。 维度查看:torch.Tensor.size() 查看当前 tensor 的维度 举个例子: >>> import 阅读全文
posted @ 2021-08-12 15:18 那抹阳光1994 阅读(832) 评论(0) 推荐(0) 编辑