Python、numpy の行列の変更に関連するいくつかの問題

1. 2 つの行列の対応する位置の大きい方(小さい方)の値を保持します。

import numpy as np
arr1 = np.random.randint(1,10,[3,3])
arr2 = np.random.randint(1,10,[3,3])
arr = arr1 * (arr1 > arr2) + arr2 * (arr2 > arr1)

2. 寸法を変更せずにチャネルを個別に選択します

import torch
inputs = torch.randn([2,3,224,224])
# 选择一个通道,维度减1
inputs1 = inputs[:,0,:,:] # shape(2,224,224)
# 选择一个通道,维度不变
inputs1 = inputs[:,0:1,:,:] # shape(2,1,224,224)

おすすめ

転載: blog.csdn.net/Huang_Fj/article/details/121982124