flatten

torch.flatten与torch.nn.flatten

torch.nn.flattentorch.nn.flatten是一个类，作用为将连续的几个维度展平成一个tensor（将一些维度合并）参数为合并开始的维度，合并结束的维度（维度就是索引，从 0 开始）开始维度默认为 1。因为其被用在神经网络中，输入为一批数据，第 0 维为batch（输入数据...
2022-01-24 21:26 阅读数 156
技术
下载桌面版
GitHub
百度网盘（提取码：draw）
Gitee
云服务器优惠
阿里云优惠券
腾讯云优惠券
华为云优惠券
站点信息
问题反馈
邮箱：[email protected]
QQ群：766591547
关注微信
©2020-2024 ioDraw All rights reserved, Privacy Policy