：）关于torch函数中dim的解释-读这篇就够了-|
摘要：
关于torch函数中dim的解释-读这篇就够了 1 dim的取值范围 1）-1的作用 0，1，2，-1. 其中-1 最后一维 即 2 0，1，2，3，-1其中-1 最后一维 即3 2）维度 0，1，2，3表示 BCHW，常在CV任务中使用。 0，1，2 表示 CHW， 常在NLP任务中使用。 3）用 阅读全文
posted @ 2023-04-08 11:18 lexn 阅读(566) 评论(0) 推荐(1) 编辑
