Как наносить нулями тензор вдоль некоторой оси (Python)
Я хотел бы заполнить тензором с нулевым значением, равным 0 вдоль выбранной оси. Например, у меня тензор r с формой (4,3,2) но меня интересует только заполнение только последних двух осей (т. (4,3,2) только матрица). Можно ли это сделать с помощью однострочного кода python?
- Обнаружение, если массив NumPy содержит хотя бы одно нечисловое значение?
- Что такое эквивалент python / numpy num2cell ()?
- Поиск пересечения кривой из полифита
- Ненужный мешгрид в 3D
- Агностическое (общее) декартово произведение размерности
- Как найти индекс значения в массиве 2d в Python?
- Преобразование строкового представления словаря в настоящий словарь
- Отображение диапазона Matplotlib yaxis с использованием абсолютных значений, а не значений смещения?
- numpy.loadtxt Пропуск нескольких строк
- Векторизация для вычисления многих расстояний
One Solution collect form web for “Как наносить нулями тензор вдоль некоторой оси (Python)”
Вы можете использовать np.pad() :
a = np.ones((4, 3, 2)) # npad is a tuple of (n_before, n_after) for each dimension npad = ((0, 0), (1, 2), (2, 1)) b = np.pad(a, pad_width=npad, mode='constant', constant_values=0) print(b.shape) # (4, 6, 5) print(b) # [[[ 0. 0. 0. 0. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 0. 0. 0.] # [ 0. 0. 0. 0. 0.]] # [[ 0. 0. 0. 0. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 0. 0. 0.] # [ 0. 0. 0. 0. 0.]] # [[ 0. 0. 0. 0. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 0. 0. 0.] # [ 0. 0. 0. 0. 0.]] # [[ 0. 0. 0. 0. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 1. 1. 0.] # [ 0. 0. 0. 0. 0.] # [ 0. 0. 0. 0. 0.]]]
- Python pandas integer YYYYMMDD для datetime
- Насколько различен тип .__ setattr__ из объекта .__ setattr__?