三つ叶个人主页

@zzhhjjjj

三つ叶

2023-01-29 14:54:13 加入 DevPress

简介

该用户还未填写简介

擅长的技术栈

未填写擅长的技术栈

可提供的服务

暂无可提供的服务

pytorch——torch.flatten() 和 torch.nn.Flatten()

flatten()函数的作用是将tensor铺平成一维torch.flatten(input, start_dim=0, end_dim=- 1) → Tensorinput (Tensor) – the input tensor.start_dim (int) – the first dim to flattenend_dim (int) – the last dim to flattensta

#pytorch #深度学习 #机器学习

到底了