Я наращиваю свои навыки с помощью PyTorch и мне интересно, есть ли способ сделать некоторые общие функции, которые возможны с Numpy. В частности, удаление столбцов, которые имеют определенные средства или конкретные стандартные отклонения.
В Numpy это достаточно легко сделать, найдя по столбцу средние / std, а затем выполните
columns_to_keep = np.concatenate( [stdColX != 0 ] )
X = X[:, columns_to_keep]
Однако, перебирая нотацию PyTorch, я не могу найти простой способ сделать это после получения столбца Means / Stds.
После того, как я
X_colStds = X.std(0)
Чтобы получить стандартное отклонение для каждого столбца в моем PyTorch Tensor, что мне делать? Я бы хотел избегать подпрыгивания вперед и назад между numpy, если я могу избежать этого.
Спасибо!