首页
Python
Java
前端
数据库
Linux
Chatgpt专题
开发者工具箱
里常专题
(done) 关于 pytorch 代码里常出现的 batch_first 到底是啥?
参考文章:https://pytorch.org/docs/stable/generated/torch.nn.utils.rnn.pad_sequence.html 首先看参考文章里的解释,如下图 从文章描述来看,当 batch_first = True 时,输出的张量的 size 是 B x T x *。当 batch_first = False 时,输出的张量的 size 是 T
阅读更多...