torch.chunk()函数 - 润新知

torch.chunk()函数

将tensor按dim方向分割成chunks个tensor块，返回的是一个元组。

示例：
相关阅读:
shell28获取命令结果的第几个参数
 在 Amazon EC2 Linux 实例上手动安装 SSM 代理 Raspbian
Trying out the Intel Neural Compute Stick 2 – Movidius NCS2
Flask
virtualenv/venv 和 pip
轻量化卷积神经网络：SqueezeNet、MobileNet、ShuffleNet、Xception
向 AWS Elastic Beanstalk 部署 Flask 应用程序
 Installing the AWS Toolkit for Visual Studio Code
Installing the AWS SAM CLI on macOS
树莓派显示器电源管理禁止屏幕休眠
原文地址：https://www.cnblogs.com/picassooo/p/14071304.html

Copyright © 2020-2023 润新知