目录
pytorch冒号 笔记
ps[:, 8:12]
从8取到12
双冒号:
ps[:, 8::2]
从8位置开始,每隔2个取一个。
python的冒号作用是一样的:
import numpy as np
if __name__ == '__main__':
x_pred = np.linspace(0, 9, num=10)
x_pred = np.arange(0, 10)
print(x_pred)
print(x_pred[2:4])
print(x_pred[2::2])
结果:
[0 1 2 3 4 5 6 7 8 9]
[2 3]
[2 4 6 8]