Vid anfallet mot Ostpreussen i januari 1945 och anfallet mot Königsberg i mars/april samma år anförtroddes Bagramjan med befälet över "Samlandgruppen", det förband som skulle utföra stormningen av staden.