в [B,1,1,S] , не используя костыли вроде tensor.unsqeeze(1).unsqueeze(1) ?
Не похоже это на костыль
Хотелось бы как нибудь через один метод (мне просто кажется что должен быть такой способ, но штудировать доку лень)
Во-первых, метод один Во-вторых, метод идеально для этого подходит, а конкретно для двух вряд ли существует (было бы странно) Поэтому, если это решает твою проблему, – используй без задних мыслей
*Один вызов метода Понял принял, спасибо
Так пойдёт ?
Обсуждают сегодня