函数定义slice(input_, begin, size, name=None)参数说明input:张量begin:切片的起始位置size:切片的大小 ... LongTensor([0, 2]) # the index vector B = torch.
確定! 回上一頁