numpy.array_split无法按预期工作

Question

我正在运行以下代码：

var aVar: AType? { get }

即。批量大小似乎是 sz = np.size(R_cap_i_val) # sz = 1055198 n_batches = sz // batch_size # batch_size = 32, n_batches = 32974 R_cap_i_ff_vals = np.array_split(R_cap_i_val, n_batches) R_cap_j_ff_vals = np.array_split(R_cap_j_val, n_batches) # len(R_cap_i_ff_vals) = 32974 # len(R_cap_i_ff_vals[0]) = 17 # len(R_cap_i_ff_vals[1]) = 17 # len(R_cap_i_ff_vals[2]) = 17 而不是17。

为什么？