Как наносить нулями тензор вдоль некоторой оси (Python)

Я хотел бы заполнить тензором с нулевым значением, равным 0 вдоль выбранной оси. Например, у меня тензор r с формой (4,3,2) но меня интересует только заполнение только последних двух осей (т. (4,3,2) только матрица). Можно ли это сделать с помощью однострочного кода python?

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 с использованием только 1,6 ГБ ОЗУ Ubuntu 32 бит в массиве Numpy
  • Могу ли я установить float128 в качестве стандартного флота-массива в numpy
  • Преобразование списка python с значениями None в массив numpy с значениями nan
  • Как файлы HEALPix FITS из карт CMB переводятся в ndarrays? Каковы координаты?
  • Как определить длину массива numpy только с одним элементом?
  • Лучший способ присоединиться / объединиться в пандах
  • Numpy: ImportError: невозможно импортировать имя TestCase
  • Проблемы с компилятором с помощью pip во время установки numpy в Windows 8.1, 7 Enterprise и 7 Home Editions
  • Python - лучший язык программирования в мире.