中的张量中获取特定行

中的张量中获取特定行