tensor

Вопрос или проблема Я хочу сделать срез 3D тензора в PyTorch. Размер 3D тензора src_tensor равен (batch, max_len, hidden_dim), и у меня есть 1D вектор индексов indices с размером (batch,). Я хочу сделать срез вдоль второго измерения src_tensor.

Вопрос или проблема Для torch.Tensor легко вычислить среднее и дисперсию, но я не могу найти способы вычислить среднее и дисперсию для вложенных тензоров. Вложенные тензоры поддерживают операцию layer_norm, которая включает операции среднего и дисперсии. Спасибо за вашу помощь!

Вопрос или проблема Полный набор данных для моей модели CNN слишком велик для 12 ГБ vRAM, которые есть у GPU, и model.fit() вызывает ошибку OOM, когда все данные загружаются. Чтобы решить эту проблему OOM, генератор данных реализован следующим образом

Вопрос или проблема Поскольку потребители научились использовать GPU для вычислений общего назначения, научились ли они использовать TPU (например, Pixel 6 Tensor) в качестве универсальных процессоров? Когда Графические процессоры общего назначения еще