Filling torch tensor with zeros after certain index ... lengths = torch .randint(1000,(10,)) ... mask[..., None]) # use mask to zero after each column.
確定! 回上一頁