4 x 5 и хочу извлечь из него матрицу 4 x 5, то какую функцию следует вызвать из pytorch?
squeeze(0)
Спасибо)
Это же не будет работать, нет?
а как вы хотите, по индексу?
x[index, :, :]
Обсуждают сегодня