2021年12月1日
PyTorch 常用函数备忘
摘要： PyTorch 常用函数备忘 # basic operation x: torch.Tensor x.shape -> torch.Size x.ndim -> int # 轴数 x.T x.numel() -> int # total size x.reshape(*shape) -> Tenso 阅读全文
posted @ 2021-12-01 22:42 Terrasse 阅读(52) 评论(1) 推荐(0) 编辑