2020年3月20日
pytorch中的size()、 squeeze()函数
摘要： size() size()函数返回张量的各个维度的尺度。 squeeze() squeeze(input, dim=None)，如果不给定dim，则把input的所有size为1的维度给移除；如果给定dim，则只移除给定的且size为1的维度。 阅读全文
posted @ 2020-03-20 00:52 Picassooo 阅读(10085) 评论(0) 推荐(0) 编辑