tf.contrib.framework.with_shape
Stay organized with collections
Save and categorize content based on your preferences.
Asserts tensor has expected shape.
tf.contrib.framework.with_shape(
expected_shape, tensor
)
If tensor shape and expected_shape, are fully defined, assert they match.
Otherwise, add assert op that will validate the shape when tensor is
evaluated, and set shape on tensor.
Args |
|---|
expected_shape
|
Expected shape to assert, as a 1D array of ints, or tensor
of same.
|
tensor
|
Tensor whose shape we're validating.
|
Returns |
|---|
|
tensor, perhaps with a dependent assert operation.
|
Raises |
|---|
ValueError
|
if tensor has an invalid shape.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2020-10-01 UTC.
[null,null,["Last updated 2020-10-01 UTC."],[],[],null,["# tf.contrib.framework.with_shape\n\n\u003cbr /\u003e\n\n|-------------------------------------------------------------------------------------------------------------------------------------------------------|\n| [View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v1.15.0/tensorflow/contrib/framework/python/framework/tensor_util.py#L232-L299) |\n\nAsserts tensor has expected shape. \n\n tf.contrib.framework.with_shape(\n expected_shape, tensor\n )\n\nIf tensor shape and expected_shape, are fully defined, assert they match.\nOtherwise, add assert op that will validate the shape when tensor is\nevaluated, and set shape on tensor.\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Args ---- ||\n|------------------|---------------------------------------------------------------------|\n| `expected_shape` | Expected shape to assert, as a 1D array of ints, or tensor of same. |\n| `tensor` | Tensor whose shape we're validating. |\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Returns ------- ||\n|---|---|\n| tensor, perhaps with a dependent assert operation. ||\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n\u003cbr /\u003e\n\n| Raises ------ ||\n|--------------|---------------------------------|\n| `ValueError` | if tensor has an invalid shape. |\n\n\u003cbr /\u003e"]]
View source on GitHub