Browse Source

add get_shape

hopefully got the types right here
pull/571/head
carb0n GitHub 5 years ago
parent
commit
8b47fff242
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      src/TensorFlowNET.Core/Operations/Operation.cs

+ 6
- 1
src/TensorFlowNET.Core/Operations/Operation.cs View File

@@ -275,6 +275,11 @@ namespace Tensorflow
return NodeDef.Parser.ParseFrom(buffer.DangerousMemoryBlock.Stream());
}
}
public TensorShape get_shape(Tensor x)
{
return x.shape;
}

/// <summary>
/// Update the input to this operation at the given index.
@@ -323,4 +328,4 @@ namespace Tensorflow
return new TF_Input(op, input_idx);
}
}
}
}

Loading…
Cancel
Save