浅记录一下,有endpoint和没有时候,这个输出很迷,不过后续都可以通过long进行限定。
value = torch.tensor(np.linspace(0, 4, num=4, endpoint=False), dtype=torch.long)
参考:
- 【Python】np.linspace用法介绍:https://blog.csdn.net/Asher117/article/details/87855493
浅记录一下,有endpoint和没有时候,这个输出很迷,不过后续都可以通过long进行限定。
value = torch.tensor(np.linspace(0, 4, num=4, endpoint=False), dtype=torch.long)
参考: